参考博客https://blog.csdn.net/wanzhen4330/article/details/81704474
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
在nvidia官网下载对应操作系统的CUDA安装包。
下载cuda9.0的安装文件 cuda_9.0.176_384.81_linux.run
执行如下命令:
sudo sh cuda_9.0.176_384.81_linux.run
然后用一本书的书角压住回车键,直到服务条款显示到100%。接着按下面的步骤选择:
accept
n(不要安装driver)
y
y
y
安装完成。
3. cudnn安装
下载对应版本的cudnn(点击进去下载),并解压。这个下载比较烦,需要注册一个账户。
cd $你的cudnn解压的目录 sudo cp ./include/cudnn.h /usr/local/cuda/include sudo cp ./lib64/lib* /usr/local/cuda/lib64/ ##复制动态链接库 cd /usr/local/cuda/lib64/ sudo ln -snf libcudnn.so.${VERSION} libcudnn.so ##强制建立软链接
4. 修改环境变量
打开主目录下的 .bashrc文件,将相关路径按下面的格式进行修改。
sudo gedit ~/.bashrc
在文本结尾添加
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH export PATH=/usr/local/cuda-9.0/bin:$PATH export CUDA_HOME=/usr/local/cuda-9.0
更新环境变量
source ~/.bashrc
5. 检查当前的cuda版本
在终端输入nvcc --version,最后一行显示Cuda compilation tools, release 9.0, V9.0.130 就说明安装成功了。