一、卸载
首先可以查看现在cuda版本
cat /usr/local/cuda/version.txt
即可显示当前cuda版本
卸载cuda时,如果是用run文件安装的,可以用如下命令卸载:
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
然后将剩余的cudnn文件也删除
sudo rm -rf /usr/local/cuda-8.0
这样就完全删除了cuda8.0
但是有些情况cuda/bin文件夹中是不含有uninstall_cuda_8.0.pl的,这时可以用另外一种方法卸载:
sudo apt-get remove cuda
sudo apt autoremove
sudo apt-get remove cuda*
然后cd /usr/local/路径
sudo rm -rf cuda-8.0
这样就完全删除了cuda8.0
二、安装cuda9.0
下载cuda9.0
https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal
执行如下命令进行安装:
sudo sh cuda_9.0.176_384.81_linux.run
然后一直按回车键,直到服务条款显示到100%
然后按accept
接着提示是否安装driver时选择n
接下来选择
y
y
y
然后就完成安装,接下来设置环境变量
配置环境变量
vim /etc/profile
在打开的文件末尾加上:
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
重启机器使生效
可以测试下cuda的samples
cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
会出现cuda的各项信息
接下来添加环境变量
vim ~/.bashrc
在打开的文件末尾加上:
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使环境变量生效
设置环境变量和动态链接库
vim /etc/profile
在打开的文件末尾加上:
exportPATH=/usr/local/cuda/bin:$PATH
创建链接文件
sudo vim /etc/ld.so.conf.d/cuda.conf
在打开的文件中写入:
/usr/local/cuda/lib64
执行sudo ldconfig
安装完cuda之后可以nvcc -V查看版本
三、安装cudnn7.0
去官网下载cudnn7.0
https://developer.nvidia.com/rdp/cudnn-download
下载后解压,进入解压后的文件夹cuda
cd cuda
sudo cp lib64/lib*/usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
然后cd到/usr/local/cuda/lib64路径,更新网络链接
cd /usr/local/cuda/lib64/
sudo chmod+r libcudnn.so.7.0.3
sudo ln -sf libcudnn.so.7.0.3. libcudnn.so.7
sudo ln -sf libcudnn.so.7libcudnn.so
sudo ldconfig
使用如下命令可查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2