1. 下载cuda10.2和cuDNN v7.6.5
1.1 下载cuda
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
1.2 下载cuDNN
从Nvidia官网登陆后,下载cuDNN
2. (可选)卸载旧驱动
sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean
3. (可选)删除旧版本cuda
cd /usr/local/cuda-10.1/bin
sudo ./cuda-uninstaller
4. (可选)手动安装Nvidia显卡驱动
插曲:我在卸载完驱动和旧版本cuda后,运行cuda_10.2.89_440.33.01_linux.run文件时选择同时安装自带的捆绑nvidia显卡驱动,但是不知道什么原因驱动安装失败,并且好像捆绑的驱动版本和我刚卸载的版本是一样的,只好手动重装驱动。
4.1 查看推荐的NVIDIA显卡驱动版本
ubuntu-drivers devices
4.2 根据4.1结果,安装NVIDIA驱动
sudo apt install nvidia-driver-XXX
注:XXX为4.1结果中显示的具体版本号
5 安装cuda&cuDNN
5.1 安装cuda v10.2
注意,因为已经手动安装适合版本的NVIDIA显卡驱动,因此在安装过程中不再勾选安装程序内捆绑安装的显卡驱动
sudo sh cuda_10.2.89_440.33.01_linux.run
安装完成后,根据安装程序提示,在.bashrc文件中添加以下两行
sudo vim ~/.bashrc
PATH="/usr/local/cuda-10.2/bin:$PATH"
LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64"
5.2 安装cnDNN v7.6.5
解压下载好的cnDNN文件
tar -zxvf cudnn-10.2-linux-x64-v7.6.5.32.tgz
安装cnDNN
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.2/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*