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

最新推荐文章于 2024-03-17 20:27:26 发布

猴猴猴猴猴猴

最新推荐文章于 2024-03-17 20:27:26 发布

阅读量1.6k

点赞数 5

分类专栏：环境配置文章标签： linux 深度学习 ubuntu

本文链接：https://blog.csdn.net/SALDJF1352/article/details/127176482

版权

环境配置专栏收录该内容

1 篇文章 0 订阅

订阅专栏

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下载界面，根据平台配置选择下载文件：
CUDNN下载：进入CUDNN8.5.0下载界面，按照如下选择下载

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选项，详情如图所示：
CUDA安装
安装完成之后会显示如下信息，需根据提示添加环境变量：
CUDA安装完成界面
键入指令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对应的安装目录中即可。具体步骤如下：

cd [文件所在路径]
tar -xvf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz

复制文件并赋予权限

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*