Ubuntu16.04+CUDA+CuDnn的踩坑实录
安装cuda.deb10.1
如果安装最新版本CUDA ,则根据官网提供的命令即可很好的完成安装,然后按照cudn和cudnn版本对应即可。
卸载cuda.deb10.1
$sudo apt-get remove cuda10.1
$sudo apt autoremove
$sudo apt-get remove
如果你安装的是.run文件,不是deb文件,则用以下命令卸载
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
卸载完 usr/local/cuda10.1 和 var/local/cuda10.1 以后,记得删掉etc/apt/sources.list.d中的cuda10.1.list,否则会出现 ” 部分索引文件失败和下载某些关于cuda10.1的文件失败 “ 的报错,导致无法安装新版本的cuda。
更换cuda版本
如果要下载其余版本的CUDA,则从图一右下角的Legacy Releases进入,上述界面下载,下载deb格式文件,按照如下命令安装:
sudo dpkg -i cuda-repo-ubuntu1604-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb (改为自己下载的文件名字)
sudo apt-key add /var/cuda-repo-10-1-local-10.1.243-418.87.00/7fa2af80.pub (改为自己下载的文件名字)
sudo apt-get update
sudo apt-get -y install cuda
配置路径
~/.bashrc 里面添加以下代码,修改成自己文件位置就好
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
查看cuda版本
命令 cat /usr/local/cuda/version.txt
或者 ncvv -v
cuDNN安装
下载链接https://developer.nvidia.com/cudnn,需要注册用户
类似于上图,按照自己安装的CUDA的版本下载cuDNN
将下载下来的文件转换成tgz的,然后解压
cp cudnn-10.0-linux-x64-v7.3.1.20.solitairetheme8 cudnn-10.0-linux-x64-v7.3.1.20.tgz
tar -xvf cudnn-10.0-linux-x64-v7.3.1.20.tgz
解压下载以后的文件,然后执行以下命令
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*