(这是原文地址,我是参考这位先生的方法安装成功的)
下载包的时候推荐QQ浏览器,我用着比其他浏览器快好多,如果下载慢的话。
版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/wanzhen4330/article/details/81704474
1.cuda的升级
(1)卸载旧cuda
卸载原来的cuda8.0(注意:不需要卸载显卡驱动,不要给自己找麻烦):
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
卸载之后,会发现 /usr/local/cuda-8.0目录下任然有文件存在,这是cudnn文件,所以还需要将cuda-8.0文件删除干净:
sudo rm -rf /usr/local/cuda-8.0
(2)安装cuda9.0
下载cuda9.0的安装文件 cuda_9.0.176_384.81_linux.run(https://developer.nvidia.com/cuda-toolkit-archive)
执行如下命令:
sudo sh cuda_9.0.176_384.81_linux.run
然后用一本书的书角压住回车键,直到服务条款显示到100%。接着按下面的步骤选择:
accept
n(不要安装driver)
y
y
y
安装完成后,设置环境变量。
(3)修改环境变量
打开主目录下的 .bashrc文件,将相关路径按下面的格式进行修改。例如我的.bashrc文件在/home/wangyuanwei下,如果没有找到,则按Ctrl+H键显示隐藏文件。
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0
在终端运行:source ~/.bashrc
(4)检查当前的cuda版本
在终端输入nvcc --version,如果显示下面的文子就说明安装成功了。
2.cudnn的升级
cudnn的升级方法与cuda的升级方法类似。
(1)删除旧版本的cudnn
sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn
(2)安装新版本的cudnn
下载所需要的cudnn版本(https://developer.nvidia.com/rdp/cudnn-archive)
解压文件,并在解压出的cuda文件夹中打开终端,执行如下命令:
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
(3)建立软链接
在/usr/local/cuda/lib64目录下打开终端,执行如下指令:
sudo chmod +r libcudnn.so.7.0.5
sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
(4)查看当前cudnn的版本
在终端输入
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出现下图所示版本信息(三个数字的意思是当前版本为7.0.5),说明安装成功。
————————————————
版权声明:本文为CSDN博主「撸起袖子就去干」的原创文章,遵循CC 4.0 by-sa版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/wanzhen4330/article/details/81704474