【最新】Ubuntu18.04配置驱动、CUDA、CUDNN_nvidia corporation device 2504

B站|公众号：啥都会一点的研究生

关于驱动

强烈建议先自行安装驱动，然后在待会安装cuda时选择不安装其附带驱动

ubuntu下查看显卡型号

lspci | grep -i vga
1

然后终端将打印

01:00.0 VGA compatible controller: NVIDIA Corporation Device 2504 (rev a1)
1

可以看到显卡型号名称是十六进制的数字代码2504，那么如何查询呢，提供如下网址
The PCI ID Repository

输入后可以得到

驱动下载

进入驱动下载地址找到对应刚刚查询的电脑配置下载驱动，具体安装步骤这里待补充（因为拿到电脑时公司已安装完驱动）

CUDA安装

当安装完驱动后CTRL + ALT + T打开终端输入

nvidia-smi
1

若安装成功则会显示如下界面

在界面上可以看到CUDA Version是11.4，下面准备安装，从CUDA下载地址下载对应于你电脑cuda的安装包，我以11.4为例

选择相应版本后可以看到官方也给出了安装命令

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run
1

下载完毕后，在文件所在处继续执行

sudo sh cuda_11.4.0_470.42.01_linux.run
1

敲黑板：选取安装项时不安装显卡驱动

网上偷的图，界面一样，按键盘下键定位至Driver，回车即取消选定，然后接着选择Install

完成后继续添加环境变量，若你使用默认bash则

sudo gedit ~/.bashrc
1

在文件末尾添加

export PATH=/usr/local/cuda-11.4/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH
1
2

注意将cuda-11.4替换为你的版本

保存后终端输入

source ~/.bashrc
1

若使用其他shell如zsh，则上述换成.zshrc就行

测试CUDA是否安装成功

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery
1
2
3

可以看到Result = PASS

此外，终端输入

nvcc -V
1

输出相应信息

安装CUDNN

进入CUDNN下载地址下载安装文件，需要用账号，自己注册一个很快

下载完毕后右键选择解压，或者在终端输入

tar -xzvf cudnn-11.4-linux-x64-v8.2.4.15.tgz
1

解压完毕后依次在终端输入如下命令，将cudnn相应文件拷贝至cuda对应位置

sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.4/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.4/lib64
sudo chmod a+r /usr/local/cuda-11.4/include/cudnn*.h /usr/local/cuda-11.4/lib64/libcudnn*
1
2
3

检查CUDNN是否安装成功

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

切换版本

比如我安装完11.4被告知团队使用11.1，为避免不必要的麻烦需要与Team保持一致，所以无需卸载，仅需按上述步骤安装11.1即可，安装完11.1的CUDA以及相应CUDNN后

sudo gedit ~/.bashrc
1

想用什么版本就将其他版本注释掉就行