下载库
确定自己所需要的cuda版本以及对应的cudnn版本。
CUDA下载链接:
https://developer.nvidia.com/cuda-toolkit-archive
cudnn下载链接:
https://developer.nvidia.cn/rdp/cudnn-download
*其中cudnn需要注册开发者账号进行下载
安装CUDA
本文以安装cuda11.7为例:
打开对应网站,选择自己所需版本,按照窗口内的指令输入即可,建议安装deb(local)
最后在自己的bash内添加cuda对应的path
gedit ~/.bashrc
添加如下命令
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
输入以下命令更新bash
source ~/.bashrc
检测cuda安装
nvcc -V
效果显著!!!
安装cudnn
此处假设已经下载好了cudnn的deb文件
根据官网安装文件指南:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html
该部分来完成安装,笔者将按照本机上的cudnn库演示安装
sudo dpkg -i cudnn-local-repo-ubuntu2004-8.5.0.96_1.0-1_amd64.deb
#后者需要修改为自己下载的cudnn库文件
复制该指令运行
sudo cp /var/cudnn-local-repo-ubuntu2004-8.5.0.96/cudnn-local-0579404E-keyring.gpg /usr/share/keyrings/
更新源,也就是把这个deb的内容添加到apt里
扫描二维码关注公众号,回复:
15091540 查看本文章
sudo apt-get update
此时我们能在apt内找到相应的库了
此时不用按照官方教程内的语句来逐字逐句输入了,直接安装这几个就行了;
sudo apt-get install libcudnn8 libcudnn8-dev libcudnn8-samples