Ubuntu20.04进行CUDA11.0及对应CUDNN安装:

基本步骤:

1.安装nvidia显卡驱动

可直接通过:软件和更新->附加驱动
选择满足CUDA版本的nvidia专有驱动->应用更改,重启即可。
(CUDA版本和显卡驱动版本对应关系:)

2.安装CUDA

2.1 CUDA下载

通过NVIDIA官网,根据硬件/系统条件,选择runfile(local),根据提供的网址下载.run文件。

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
sudo sh cuda_11.0.2_450.51.05_linux.run

PS:
官方下载方式为通过终端运行wget下载。但笔者建议复制网址,通过网页下载,笔者通过wget下载时,到最后几秒出现段错误(核心已转储)问题,终止下载,重新运行会重新从头下载。

2.2 CUDA安装

下载完成后执行以下语句:

sudo sh cuda_11.0.2_450.51.05_linux.run

选择continue,输入accept,根据需要选择是否需要安装相应nvidia驱动(PS:在第1步安装了驱动则建议在这里取消(空格)driver安装,不然可能安装失败。)其他默认(均选中)即可。
选择install回车。
安装完成后:
通过gedit ~/.bashrcvim ~/.bashrc等,打开~/.bashrc文件,在最后添加:

export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

保存关闭,并source ~/.bashrc

3.安装CUDNN

参考NVIDIA cuDNN v8 deb方法安装和卸载教程(Linux/Ubuntu)

3.1 CUDNN下载

官网找到需要的cudnn版本,下载cuDNN Runtime xxx/ cuDNN Developer xxx/ cuDNN Code xxx三个.deb文件(下载后为libcudnn8_8xxx.deb / libcudnn8-debxxxx.deb / libcudnn8-samplesxxx.deb)。

3.2 CUDNN安装

依次执行以下语句进行安装:

sudo dpkg -i llibcudnn8_8.0.5.39-1+cuda11.0_amd64deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.0_amd64.deb

3.3 CUDNN测试

通过以下语句进行测试:

cp -r /usr/src/cudnn_samples_v8/ $HOME
cd  ~/cudnn_samples_v8/mnistCUDNN
sudo make clean 
sudo make
sudo ./mnistCUDNN

出现Test passe则表明cuDNN已通过程序测试,完成安装!
在这里插入图片描述


版权声明:本文为qq_43326927原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
原文链接:https://blog.csdn.net/qq_43326927/article/details/122584522