在Ubuntu系统中,使用nvidia-smi
命令可以看到当前GPU信息,在右上角可以看到CUDA Version,意思是最大支持的CUDA版本号。
拟安装:
cuda 11.8
cuDNN v8.9.0 (for CUDA 11.x)
安装CUDA
打开官网:https://developer.nvidia.com/cuda-toolkit-archive
选择CUDA 11.8,如下图选择“Lunix”及对应的架构、系统、版本和安装形式:
选择runfile(local)形式,网站会给我们安装的命令:
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run sudo sh cuda_11.8.0_520.61.05_linux.run
依次在终端中输入这两条命令!
需要注意的事项有下面几个:
一、由于我们之前以及安装过驱动了,这次不要再安装驱动了。
在Driver那里按空格键去掉勾选!
二、选项最后面的核心也不要去勾选(它默认是非勾选状态的)
三、安装过程中需要手动输入“accept”接受;
配置环境变量
打开终端,配置环境变量:
export PATH=$PATH:/usr/local/cuda-11.8/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
输入:wq
退出。
打开终端输入:
source ~/.bashrc
即可完成环境变量的配置!
为了验证有没有安装成功,可以在终端输入nvcc -V
即可查看CUDA版本:
安装cudnn
下载安装文件
打开并登陆 https://developer.nvidia.com/rdp/cudnn-archive
选择下载 cuDNN v8.9.0 (for CUDA 11.x),
再选择 Local Installer for Ubuntu22.04 x86_64 (Deb),
就开始下载了。下载完后,建议把deb文件移动到home主目录
这里需要注册账户并且登录才能下载!
文件下载好之后剪切到主文件夹!
安装
然后在终端分别输入如下的命令进行安装!
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.0.131_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.0.131/cudnn-local-2063C34E-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install libcudnn8=8.9.0.131-1+cuda11.8
sudo apt-get install libcudnn8-dev=8.9.0.131-1+cuda11.8
sudo apt-get install libcudnn8-samples=8.9.0.131-1+cuda11.8
sudo apt-get install libfreeimage3 libfreeimage-dev
验证是否安装成功
cp -r /usr/src/cudnn_samples_v8/ $HOME cd $HOME/cudnn_samples_v8/mnistCUDNN make clean && make ./mnistCUDNN
当看到“Test passed!”就意味着安装成功了!
原创文章,作者:朋远方,如若转载,请注明出处:https://caovan.com/zaiubuntu-22044anzhuangcudahecudnn/.html