目录
安装CUDA
下载CUDA,从https://developer.nvidia.com/cuda-downloads,下载对应版本的cuda安装包,我下载的是deb(local)版,
各个版本下载:
https://developer.nvidia.com/cuda-toolkit-archive
下载 runfile(local)
sudo sh cuda_11.8.0_520.61.05_linux.run
安装完成后,重启一下;
cudnn deb格式安装
chmod +x ./cudnn-local-repo-ubuntu2004-8.9.7.29_1.0-1_amd64.deb
安装软件包:使用以下命令开始安装.deb文件:
sudo dpkg -i /path/to/your_file.deb
有时会有提示:
sudo cp /var/cudnn-local-repo-ubuntu2004-8.9.7.29/cudnn-local-CD2C2DD4-keyring.gpg /usr/share/keyrings/
执行一下,再执行:
sudo dpkg -i /runxxx.deb
设置环境变量
在打开的文件中加入如下两句话
vim /etc/profile
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
保存后,使环境变量立即生效,
source /etc/profile