ubuntu16.04卸载和安装CUDA以及CUDNN

1.CUDA卸载

卸载原来的cuda9.0(不需要卸载显卡驱动):

sudo /usr/local/cuda-8.0/bin/uninstall_cuda_9.0.pl

卸载之后, /usr/local/cuda-9.0目录下任然有文件存在,这是cudnn文件,还需要将cuda-9.0文件删除干净:

sudo rm -rf /usr/local/cuda-9.0

2.CUDA安装

(1)下载cuda10.1的安装文件 cuda_10.1.168_418.67_linux.run,将下载好的文件放入home文件夹中。
执行如下命令:

sudo chmod +x cuda_10.1.168_418.67_linux.run
sudo sh cuda_10.1.168_418.67_linux.run

然后按住回车键,直到服务条款显示到100%。接着按下面的步骤选择:输入accept,在提示是否安装NVIDIA驱动,选择N。后面的其他提示都选择默认或者y。

即:
accept

n(不要安装driver)

y

y

y
安装完成后,设置环境变量。

(2)修改环境变量
打开主目录下的 .bashrc文件,将相关路径按下面的格式进行修改。或者使用以下命令打开:

vim .bashrc

在.bashrc文件的末尾添加:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

最后在终端运行:

source ~/.bashrc

(3)检查当前的cuda版本
在终端输入

nvcc --version

在这里插入图片描述

3.CUDNN卸载

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn

4.CUDNN安装

下载好适合cuda版本的cudnn的deb文件或tgz文件,以cudnn-10.1-linux-x64-v7.6.4.38.tgz的安装为例。进入压缩包所在的目录分别执行以下命令:
(1)首先将tgz文件解压

tar -xzvf cudnn-10.1-linux-x64-v7.6.4.38.tgz

(2)解压文件,并在解压出的cuda文件夹中打开终端,执行如下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chomd a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 

到此cudnn安装完毕。

(3)查看当前cudnn的版本,在终端输入:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果出现下图所示版本信息(三个数字的意思是当前版本为7.6.4),说明安装成功。
在这里插入图片描述

发布了5 篇原创文章 · 获赞 5 · 访问量 223

猜你喜欢

转载自blog.csdn.net/weixin_43450885/article/details/105031353