1. 查看显卡驱动版本,找对应cuda版本关系
当前的环境为:ubuntu18.04,nvidia driver 460.27.04
满足CUDA11.0 的 显卡驱动要求,注意到CUDA 11.0 (11.0.3)
2. 下载CUDA安装包
注意到CUDA 11.0 (11.0.3),以及CUDA和显卡驱动对应关系。
进入官网https://developer.nvidia.com/cuda-toolkit-archive
CUDA Toolkit 11.0 Update1 (Aug 2020), Versioned Online Documentation
CUDA Toolkit 11.0 (May 2020), Versioned Online Documentation
上面的update1 和 下面的 版本号不一样:
上面的是11.0.3 , 下面的是11.0.2
(PS:2021年7月9日21点31分,本来11.0.2版本已经下完了,还是选择了11.0.3版本,感觉这两个版本应该差不多)
3. 下载cudnn
进入官网https://developer.nvidia.com/rdp/cudnn-archive
选择cudnn版本8.1.1
4. 安装CUDA
安装nvidia显卡驱动时候应该已经把nouveau驱动加入到黑名单里了,这里检查一下,即可
lsmod | grep nouveau
如果没有输出,说明已经禁用成功了。
cd到.run文件所在的目录;
更改权限:chmod 777 cuda_11.0.3_450.51.05_linux.run
执行文件:
sudo ./cuda_11.0.3_450.51.05_linux.run --no-opengl-libs
sudo sh cuda_8.0.61_375.26_linux.run --no-opengl-libs
不选择驱动driver!
查看cuda版本
cat /usr/local/cuda/version.txt