之前电脑有安装cuda8,如果是第一次安装cuda,需要关闭第三方显卡驱动,参照https://blog.csdn.net/TFZ941214/article/details/103022779
准备cuda与cudnn的文件
cudnn下载地址https://developer.nvidia.com/rdp/cudnn-download#a-collapse765-101
名称与内容长下面的样子
note:
cuda历史版本下载地址(https://developer.nvidia.com/cuda-toolkit-archive)
cudnn下载地址(https://developer.nvidia.com/rdp/cudnn-archive)
1.卸载原来的cuda8
sudo apt-get autoremove --purge cuda
可能会出现警告(dpkg:警告:卸载 cuda-license-8-0 时,目录 /usr/local/cuda-8.0 非空,因而不会删除该目录)
进行手动删除
cd /usr/local
sudo rm -rf cuda-8.0
2.删除原来的cudnn
需要知道自己当时的安装位置,我安装的位置是/usr/local/include以及/usr/local/lib
cd /usr/local/include/
sudo rm cudnn.h
cd /usr/local/lib/
sudo rm libcudnn.*
3.安装cuda10.1与cudnn
参考之前的安装方法Step2与Step3https://blog.csdn.net/TFZ941214/article/details/103022779
Step 2 安装cuda
进入到cuda .deb安装位置
sudo dpkg -i cuda-repo-ubuntu1604-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
sudo apt-get update
sudo apt-get install -y cuda
安装完毕之后,打开bashrc文件,将cuda路径写入
sudo gedit ~/.bashrc
文件打开后,后面添加
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=:/usr/local/cuda/lib64
保存并关闭,运行
source ~/.bashrc
sudo reboot
Step 3 安装配置nvidia cuDNN 加速模型运算
解压cudnn压缩包,解压完后的名字应该为cuda
进入cudnn解压的cuda文件夹
cd cuda
sudo cp include/cudnn.h /usr/local/include
sudo cp lib64/libcudnn.* /usr/local/lib
链接cuDNN的库文件(核对是不是7.6.5 / 7,不是的话修改为自己的)
sudo ln -sf /usr/local/lib/libcudnn.so.7.6.5 /usr/local/lib/libcudnn.so.7
sudo ln -sf /usr/local/lib/libcudnn.so.7 /usr/local/lib/libcudnn.so
sudo ldconfig –v
到这就装完了,检查CUDA以及cudnn安装情况以及版本号
cat /usr/local/cuda/version.txt
cat /usr/local/include/cudnn.h | grep CUDNN_MAJOR -A 2
一切顺利的话,会出现如下界面