Ubuntu系统安装Nvidia显卡驱动、Cuda、Cudnn、Pytorch、Tensorflow

如果你的机器显卡是集成显卡，或者是老旧版本，那么不支持GPU加速，只能使用CPU版本的Pytorch/Tensorflow ！

本文的前提是你有一块好的Nvidia显卡。

1.如何查看电脑的显卡型号

在windows系统上，查看显卡型号的方法如下：

打开设备管理器->显示适配器

在ubuntu系统上，查看显卡型号的方法如下：

#terminal输入
lspci | grep -i vga
#将会得到序列号，我的是2489
01:00.0 VGA compatible controller: NVIDIA Corporation Device 2489 (rev a1)
#然后打开下面的链接，用刚刚的序列号,在咨询框中查询即可
http://pci-ids.ucw.cz/mods/PC/10de?action=help?help=pci

我的显卡是Nvidia-RTX 3060Ti

 2.如何查看电脑当前的显卡驱动版本

 一般电脑安装好Linux/Ubuntu系统以后，有的系统捆绑了Nvidia显卡驱动，在终端输入以下内容即可查看

nvidia-smi

如果出现了以下内容，则表示已经安装了Nvidia显卡驱动；

如果什么都没有，则需要重新安装！

3.如何安装Nvidia显卡驱动

在进行这一步骤之前，你应该分清楚: 你的电脑之前有没有安装过Nvidia显卡驱动 还是 已经安装了想要换一个版本。因为这两个执行策略不一样，如果没有安装过，请看3.1；如果是后者，请看3.2。

3.1 安装Nvidia显卡驱动

推荐五种方法安装Nvidia显卡驱动，A,B,C,D，E。方法C稍微复杂一些。方法D是cuda和Nvidia显卡驱动捆绑安装。安装完成后，都需要重启系统才能生效。

A. 系统软件更新，进行安装

B.终端命令行进行安装

sudo ubuntu-drivers autoinstall 
sudo apt install nvidia-470 #ubuntu 18.04

sudo ubuntu-drivers devices  # 查看推荐驱动版本
sudo apt install nvidia-driver-550 

上述第一条命令，亲测有效！

C.手动下载驱动文件进行安装

首先去 英伟达官网，下载正确的驱动文件：Official Drivers | NVIDIA