source:https://www.bilibili.com/video/BV1bW4y197Mo/?spm_id_from=333.999.0.0&vd_source=cb51e6e50c82c7f849cf3713e202e16c
选择CUDA版本
根据pytorch选择CUDA版本
根据网页显示选择CUDA11.8
登录官网
选择版本:
输入代码进行安装:
在终端输入代码进行安装
gedit ~/.bashrc
将下面的代码粘贴到代码的最下方
export PATH=/usr/local/cuda-11.8/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:${LD_LIBRARY_PATH}
输入代码进行验证是否成功:
source ~/.bashrc
nvcc --version
显示结果为:
验证成功
CUDNN
登录CUDNN官网,第一次登陆需要注册账号进行登录
按照自己的Linux型号进行选择(在设置-关于中查看):
下载后在该文件的终端上输入以下代码:
sudo apt-get install zlib1g
sudo dpkg -i cudnn-local-repo-${distro}-8.x.x.x_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y
sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y
如何验证cudnn?
cudnn验证:
1. cp -r /usr/src/cudnn_samples_v8/ $HOME
2. cd $HOME/cudnn_samples_v8/mnistCUDNN
3. make clean && make
4. /mnistCUDNN
在第三步出现了这个问题:
解决方式:
sudo apt-get install libfreeimage3 libfreeimage-dev
如果出现Test passed则说明通过了