Ubuntu20.04+RTX3090+CUDA11.4+CUDNN8.5.0安装与卸载

目录

  • CUDA&CUDNN卸载
  • CUDA&CUDNN下载与安装
    • CUDA&CUDNN下载
    • CUDA&CUDNN安装

CUDA&CUDNN卸载

CUDA卸载有两种方式,一种是执行CUDA自带的卸载脚本,该脚本位于/usr/local/cuda-xxx/bin目录下,使用如下指令卸载:

  • CUDA10.1及以下
sudo /usr/local/cuda-xxx/bin/uninstaller_cuda_x.x.pl
  • CUDA10.1及以上
sudo /usr/local/cuda-xxx/bin/cuda-uninstaller

如果不确定自己的CUDA版本,可以去脚本存储目录下查看文件名。

卸载完成后,利用如下指令删除遗留的文件即可:

rm -rf /usr/local/cuda-xxx

CUDA&CUDNN下载与安装

CUDA&CUDNN下载

  • CUDA下载:进入CUDA11.4下载界面,根据平台配置选择下载文件:
    Ubuntu20.04+RTX3090+CUDA11.4+CUDNN8.5.0安装与卸载_第1张图片
  • CUDNN下载:进入CUDNN8.5.0下载界面,按照如下选择下载
    Ubuntu20.04+RTX3090+CUDA11.4+CUDNN8.5.0安装与卸载_第2张图片

CUDA&CUDNN安装

  • CUDA安装

进入文件所在目录,执行:

sudo sh cuda_11.4.4_470.82.01_linux.run

若提示Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing选择Continue继续安装,如果之前安装过nvidia驱动的话,安装选项中取消Driver选项,详情如图所示:
Ubuntu20.04+RTX3090+CUDA11.4+CUDNN8.5.0安装与卸载_第3张图片
安装完成之后会显示如下信息,需根据提示添加环境变量:
Ubuntu20.04+RTX3090+CUDA11.4+CUDNN8.5.0安装与卸载_第4张图片
键入指令sudo vim ~/.bashrc打开配置文件,根据提示中指定的路径配置环境变量:

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

这里我们将cuda-11.4目录改成了cuda,实质上cuda为系统为cuda-11.4建立的软连接(类似windows中的快捷方式),因此两个目录最终指向相同的文件,不过设置为cuda目录在重装或切换多版本CUDA时会更加方便,它可以避免繁琐的修改配置文件中的版本号,只要使用ln指令更改软链接就好了。

执行 source ~/.bashrc更新配置文件完成安装,并执行nvcc -V检查安装结果,显示如下信息则表示安装成功。
NVCC版本展示

  • CUDNN安装

作为CUDA的补充库,安装时只需将CUDNN解压缩,并将其中的文件复制到CUDA对应的安装目录中即可。具体步骤如下:

  1. 进入CUDNN目录并解压缩
cd [文件所在路径]
tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz
  1. 复制文件并赋予权限
cd cudnn-linux-x86_64-8.5.0.96_cuda11-archive #进入解压后的文件夹
sudo cp include/cudnn* /usr/local/cuda-11.4/include
sudo cp lib/libcudnn* /usr/local/cuda-11.4/lib64
sudo chmod a+r /usr/local/cuda-11.4/include/cudnn* /usr/local/cuda-11.4/lib64/libcudnn*

键入指令cat /usr/local/cuda/include/cudnn_version.h |grep ^#检验安装结果,显示如下结果表明配置成功:
Ubuntu20.04+RTX3090+CUDA11.4+CUDNN8.5.0安装与卸载_第5张图片

你可能感兴趣的:(环境配置,linux,深度学习,ubuntu)