文章目录
-
- 安装CUDA
- 安装cudnn
-
- 下载安装文件
- 安装
- 验证是否安装成功
在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
成功: