RTX3090 GPU环境配置

前言

这篇文章中安装的版本都是根据我自己的电脑的情况选择的，有需要的可以根据实际情况选择，
我的系统上Ubuntu20.04，显卡是RTX3090。跟我一样的话就大胆copy，不一样的话就注意点，名字上会有区别。

安装Nvidia显卡驱动

nvidia-smi

---

方法1：
打开Software & Updates > Additional Drivers选择第一个Using NVIDIA driver metapackage from nvidia-driver-460(proprietary,tested)然后Apply Changes，就开始下载驱动了。
安装完成是这样的：

---

方法2：
或者直接到官网下载下来

驱动程序下载

下载对应驱动，我这里的驱动是NVIDIA-Linux-x86_64-460.56.run

chmod +x NVIDIA-Linux-x86_64-460.56.run
sudo sh NVIDIA-Linux-x86_64-460.56.run

一路确定OK

---

装完驱动之后，重启电脑：

这样驱动就安装成功了
可以更新一下软件：

sudo apt update
sudo apt upgrade

安装CUDA

nvcc -V
# nvcc: NVIDIA (R) Cuda compiler driver
# Copyright (c) 2005-2020 NVIDIA Corporation
# Built on Mon_Oct_12_20:09:46_PDT_2020
# Cuda compilation tools, release 11.1, V11.1.105
# Build cuda_11.1.TC455_06.29190527_0

这是已经装好了CUDA11.1，但是这里的版本和上面的截图里CUDA Version: 11.2不太一样

CUDA Toolkit Archive
CUDA Toolkit 11.2

wget https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux.run
chmod +x cuda_11.2.1_460.32.03_linux.run
sudo sh cuda_11.2.1_460.32.03_linux.run

因为前面已经可以看到，我之前是已经安装了CUDA的，这里建议先删除在安装。

我这里就把所有东西都给删了，结果发现刚装的驱动也没了。又走了一遍安装驱动。所以这一步建议在最开始就删除。

sudo apt-get remove --purge nvidia*

可是还是一样，那就不管了，直接Continue
输入accept
第一项按空格键反选，因为驱动已经安装过了，版本也一致。
然后移动到最后一行的Install
选择yes，然后根据上图提示需要配置环境变量：

gedit ~/.bashrc

在文件最后加入以下语句：

# 这里的cuda路径不加版本号，以便后续升级cuda版本的时候不需要重新配置
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

然后使其生效：

source ~/.bashrc

安装的CUDA版本统一被安装到/usr/local文件夹下:

ls -l /usr/local

系统是通过/usr/local/下的cuda符号链接来识别CUDA当前指定的CUDA版本

可以看到路径下有cuda-11.1和cuda-11.2两个版本，但是使用的是哪个呢？就看上面这个cuda指向哪里cuda -> /usr/local/cuda-11.2/

cd /usr/local/
# 删除原来的软链接
sudo rm cuda
# 新建新的软链接
sudo ln -s /usr/local/cuda-11.2 /usr/local/cuda

查看版本：

nvcc -V
# nvcc: NVIDIA (R) Cuda compiler driver
# Copyright (c) 2005-2021 NVIDIA Corporation
# Built on Thu_Jan_28_19:32:09_PST_2021
# Cuda compilation tools, release 11.2, V11.2.142
# Build cuda_11.2.r11.2/compiler.29558016_0

安装cuDNN

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

cuDNN Download
下载cuDNN需要注册账号

选择适合CUDA11.2的第一项，在它下面选择cuDNN Library for Linux (x86_64)下载：

这是个.tgz的压缩包。直接解压就可以用了。不过这个包还挺大的，压缩包就有1.2G。

下载完成后将其解压，命令如下：

# -C解压到指定文件夹
tar -xzvf cudnn-11.2-linux-x64-v8.1.1.33.tgz -C cudnn-112

使用以下两条命令复制这些文件到cuda目录下/usr/local/cuda：

cd cudnn-112
sudo cp cuda/lib64/* /usr/local/cuda/lib64/
sudo cp cuda/include/* /usr/local/cuda/include/

copy完成之后，查看cuDNN的版本信息：

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

这样就全部安装完毕了。

---

1. 安装驱动可以不独立安装，在安装cuda时一起安装
2. 安装cuDNN其实就是将一个压缩包中的文件放到cuda安装路径的对应位置