欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-05-27 13:30:06
...

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

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

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),说明安装成功。
ubuntu16.04卸载和安装CUDA以及CUDNN

相关标签: cuda ubuntu linux