ubuntu20.04+win10双系统安装Nvidia V100 GPU驱动+cuda11.1+cudnn

1. 系统：ubuntu20.04+window10
2. 硬件配置：2Nvidia V100 4Nvidia 3090, CPU未知
   参考链接1；参考链接2；参考链接3； 参考链接4

一、Nvidia驱动安装：

1、系统安装完成后，查询GPU是否存在，型号信息

sudo lspci |grep -i nvidia
#显示GPU信息

2、 查看推荐驱动
带“recommended”选项的，即为系统的推荐版本，图是借别人的，我的是470

 ubuntu-drivers devices

3、禁止ubuntu自动更新系统
安装驱动前一定要禁止ubuntu自动更新系统，一定要关闭！！！后期自动更新，会导致驱动不能使用！！关闭ubuntu自动更新升级

1）打开系统设置

  2）进去后打开“Software & Updates（软件和更新）”，如下所示。

3）关闭 自动更新按如下界面进行设置

在界面上的“Update（更新）”选项卡中，如果要 开启 自动更新按如下界面进行设置

4）中间如果弹出验证框，输入登陆密码就可以了。

	5）禁止更新Ubuntu系统内核，上图中“有新版本时通知我”选“从不”。

关闭更新后。继续nvidia驱动安装！

4、 通过《软件和更新》在附加驱动选项中添加了驱动：
选择推荐-应用更改，即可自动安装推荐驱动，自动安装nvidia驱动！！很高效！！
5、查看驱动是否安装成功

nvidia-smi

驱动安装成功！！

二、安装CUDA+CUDNN

1. 检查自己电脑支持的cuda

nvidia-smi

2. 下载cuda11.1.0

1) wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run
2)sudo sh cuda_11.1.0_455.23.05_linux.run

3. 按照以下步骤安装

Continue

accept

取消Driver

图是之前 11.4的，不影响取消driver即可！

4. CUDA安装完成

5. 添加CUDA环境变量

gedit ~/.bashrc
#添加：
# 将下面的cuda-11.1替换为你的cuda版本，其他不变，如果不知道自己安装的是哪个版本，就去/usr/local/文件夹下找一下（因为代码调试原因，将cuda11.4换为11.1）
export PATH=$PATH:/usr/local/cuda-11.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.1/lib64
##我的bashrc里面这两条都有
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.1
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
export PATH=$PATH:/usr/local/cuda-11.1/bin

#保存退出
source ~/.bashrc

6. 测试CUDA

nvcc -V
显示cuda版本即可

7. 安装cudnn

下载：
https://developer.nvidia.com/rdp/cudnn-download
解压：
tar zxvf cudnn-11.4-linux-x64-v8.2.4.15.tgz

# 将文件复制到cuda对应的文件夹下
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
# 赋予文件执行权限
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

8. 检查cuda+cudnn

#在终端输入
import torch
print(torch.version.cuda)
	#11.1
print(torch.backends.cudnn.version())
	#8200
torch.cuda.is_available()
	#True

到此，说明 GPU 驱动、CUDA、cuDNN 已经完全安装完成。