最近在配一个深度学习环境,先后安装了CUDA10.0,CUDA8.0,CUDA10.0,CUDA9.0,每次安装卸载都要百度一番。
在此我好好整理了自己用到的指令。
一、安装 显卡驱动
已安装好显卡驱动的可以直接跳到第二步
可以参考这篇博客,亲测可用:https://blog.csdn.net/xunan003/article/details/81665835
二、安装 CUDA
本文以CUDA9.0为例
1.下载CUDA
CUDA官方下载地址:
https://developer.nvidia.com/cuda-toolkit-archive
如图下载.run格式的runfile
2.安装CUDA
下载后打开终端,输入如下指令(具体文件名换成自己刚刚下载的.run文件的文件名)
sudo sh cuda_9.0.176—384.81_linux.run
之后按会有一大段让你阅读的文字,按住回车一直到末尾
依次输入
accept
n
y
y
y
其中第一个n表示:不安装显卡驱动(driver)
第1和第3个y后面需要填安装路径,直接按回车默认路径即可
3.配置环境变量
修改/etc/profile
打开gedit对文件进行修改
sudo gedit /etc/profile
在文件末尾添加(根据自己的CUDA版本进行粘贴)(操作系统为64位的情况下)
下面3个根据自己版本直接粘贴
# CUDA-8.0
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
# CUDA-9.0
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
# CUDA-10.0
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并关闭文件后继续在命令行输入
source /etc/profile
修改 ~/.bashrc
sudo gedit ~/.bashrc
在文件末尾添加
下面3个根据自己版本直接粘贴
# CUDA-8.0
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
# CUDA-9.0
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
# CUDA-10.0
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存并关闭文件后继续在命令行输入
source ~/.bashrc
检验是否成功
nvcc -V
OK.
三、安装Cudnn
下载cudnn
cudnn官方下载网址:https://developer.nvidia.com/rdp/cudnn-archive
根据CUDA版本下载,我下载的是 cudnn-9.0-linux-x64-v7.5.1.10.tgz
将其解压
tar -xzvf cudnn-9.0-linux-x64-v7.5.1.10.tgz
将cudnn的文件复制到CUDA的安装目录下,并赋予权限(注意目标路径根据CUDA版本路径名不同)
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*
检验是否安装成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
OK.
四、如何卸载CUDA和cudnn
卸载CUDA(9.0)
sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl
删除之前复制进去的cudnn的相关文件以及整个文件夹
sudo rm -rf /usr/local/cuda-9.0
----------
对我帮助很大的参考博客:
https://blog.csdn.net/wanzhen4330/article/details/81704474
https://www.cnblogs.com/liuwenhua/p/11521668.html
https://blog.csdn.net/chengyq116/article/details/95513949
https://blog.csdn.net/weixin_42279044/article/details/83181686