在Ubuntu系统中,使用nvidia-smi命令可以看到当前GPU信息,在右上角可以看到CUDA Version,意思是最大支持的CUDA版本号。
上一篇文章已经安装了显卡驱动,这次继续安装CUDA、cudnn。
在这里插入图片描述
拟安装:
cuda 11.7
cuDNN v8.9.0 (for CUDA 11.x)

安装CUDA

打开官网:https://developer.nvidia.com/cuda-toolkit-archive
选择CUDA 11.7,如下图选择“Lunix”及对应的架构、系统、版本和安装形式:
在这里插入图片描述

选择runfile(local)形式,网站会给我们安装的命令:
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

依次在终端中输入。

第一条命令:
在这里插入图片描述

第二条命令:
注意!由于我们之前以及安装过驱动了,这次不要再安装驱动了。

输入accept
在这里插入图片描述

在Driver这里按ENTER勾掉!
然后选择Install
在这里插入图片描述

在这里插入图片描述

最后,打开终端,配置环境变量:

vim ~/.bashrc

最后添加:

export PATH=$PATH:/usr/local/cuda-11.7/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.7/lib64

输入:wq退出。
打开终端输入:

source ~/.bashrc

完成!


最后终端输入nvcc -V即可查看CUDA版本:
在这里插入图片描述

安装cudnn

下载安装文件

打开并登陆 https://developer.nvidia.com/rdp/cudnn-download
选择下载 cuDNN v8.9.0 (for CUDA 11.x),
再选择 Local Installer for Ubuntu22.04 x86_64 (Deb),
就开始下载了。下载完后,建议把deb文件移动到home主目录。

在这里插入图片描述


安装

打开上方的 Installation Guide 可查看安装方法。
其中我们已经安装了显卡驱动程序和CUDA。
如果系统还没有安装zlib包,可先sudo apt-get install zlib1g安装一下。

在这里插入图片描述

安装步骤:
1.打开deb文件存放目录,右键打开终端,输入(deb文件名和你下载的一致):
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.0.131_1.0-1_amd64.deb

2.Import the CUDA GPG key.
在这里插入图片描述

输入:
sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.0.131/cudnn-local-2063C34E-keyring.gpg /usr/share/keyrings/

3.刷新库:
sudo apt-get update

4.Install the runtime library.

注意,这里的libcudnn8和cuda版本的配对是指定的,可通过apt-cache policy libcudnn8命令查看。
我这里应该使用libcudnn8=8.9.0.131-1+cuda11.8

输入:
sudo apt-get install libcudnn8=8.9.0.131-1+cuda11.8

5.Install the developer library.

sudo apt-get install libcudnn8-dev=8.9.0.131-1+cuda11.8

6.Install the code samples and the cuDNN library documentation.

sudo apt-get install libcudnn8-samples=8.9.0.131-1+cuda11.8

验证是否安装成功

在这里插入图片描述

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

注:如果上面的make命令提示缺少FreeImage.h,运行:sudo apt-get install libfreeimage3 libfreeimage-dev

成功:
在这里插入图片描述

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐