Nvidia显卡驱动

禁用自带显卡启动

注意！在安装NVIDIA驱动以前需要禁止系统自带显卡驱动nouveau：可以先通过指令lsmod | grep nouveau查看nouveau驱动的启用情况，如果有输出表示nouveau驱动正在工作，如果没有内容输出则表示已经禁用了nouveau。

如果没有禁用ubuntu自带的显卡驱动，需要以下操作

在终端输入，打开blacklist.conf文件

sudo gedit /etc/modprobe.d/blacklist.conf

在文件的末尾添加以下两行，并保存

blacklist nouveau
options nouveau modeset=0

在终端中输入

sudo update-initramfs -u  // 应用更改

安装nvidia显卡驱动

打开系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改。该界面会自动根据电脑上的GPU显示推荐的NVIDIA显卡驱动。

一般选则nvidia driver metapackage nvidia-driver-xxx后面数字更大的一个，其最高支持的CUDA版本也会更高
安装完成后可以通过以下指令验证

nvidia-smi

其中CUDA version 是指支持的最高的cuda版本

CUDA安装

下载

从nvidia官网上下载对应版本的cuda，注意cuda，cudnn，tensorflow版本要对应，这里推荐cuda11.7+cudnn8.8.0
版本对应可以从这里查到
Build from source on Windows | TensorFlow (google.cn)
安装cuda前先安装一些依赖

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

CUDA下载网址
CUDA Toolkit Archive | NVIDIA Developer

安装

选择完成后按照下方生成的指令执行

wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
sudo sh cuda_11.7.0_515.43.04_linux.run

运行上面指令后，会弹出如下界面，点击Continue，然后再输入accept。

在弹出的界面中通过Enter键，取消Driver和450.51.05的安装，然后点击Install，等待 安装完成。

配置CUDA的环境变量

在./bashrc中添加以下内容，注意要和自己cuda实际版本，路径相同

export PATH=$PATH:/usr/local/cuda/bin  
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64  
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

cuda检测

进入NVIDIA CUDA示例包，其位于/home/xxx/NVIDIA_CUDA-11.7_Samples内，在该文件夹下打开终端，并输入make。然后进入1_Utilities/deviceQuery文件夹，并在终端执行./deviceQuery 命令，如下result=PASS则表示安装成功。

cuDNN安装

下载

在nvidia官网下载与cuda对应版本的压缩包，11.x表示对11.版本都支持。
cuDNN Archive | NVIDIA Developer

安装

解压下载后的压缩包

tar -zxvf cudnn-11.0-linux-x64-v8.8.0.tgz

复制cuda文件夹下的文件到/usr/local/cuda-11.7/lib64/和/usr/local/cuda-11.7/include/中。

cp cuda/lib64/* /usr/local/cuda-11.7/lib64/
cp cuda/include/* /usr/local/cuda-11.7/include/

通过以下命令查看cudnn版本

cat /usr/local/cuda-11.7/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

cuda以及cudnn的卸载

进入到/usr/local/cuda-11.7/bin目录下，然后打开终端，输入

sudo ./cuda-uninstaller

输入命令后，弹出如下界面，通过回车键选中三个选项，最后选中Done。执行完下面指令后，cuda文件就删除了

最后，在终端输入命令

sudo rm -rf /usr/local/cuda-11.7

Pytorch，torchvision安装

版本问题

torch，cuda版本对应
Previous PyTorch Versions | PyTorch
torch，torchvision版本对应
GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision

安装

推荐采用 torch 1.13.0 + torchvision 0.14.0 + cuda 11.7
安装可以通过conda指令安装，也可以通过pip指令安装，这里采用pip
在终端输入

pip install torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0

其中torchaudio根据需要选择是否安装
官方pytorch历史版本安装教程连接
https://pytorch.org/get-started/previous-versions/
测试
在终端输入python进入python环境，输入以下指令查看torch，torchvision版本

import torch
import torchvision
torch.__version__
torchvision.__version__

+cu11.7表示cuda版本是11.7
查看torch是否可以调用cuda，GPU

torch.cuda.is_available()