Ubuntu/Linux 安装GPU 驱动&检测（PyTorch准备）

1. 操作整理

https://gist.github.com/MihailCosmin/affa6b1b71b43787e9228c25fe15aeba
https://www.cherryservers.com/blog/install-cuda-ubuntu
https://zhuanlan.zhihu.com/p/630067986

2. 方法一

2.1. 检查GPU是Nvidia

lspci | grep -i nvidia

2.2.0 非必要：如已有驱动，卸载已有驱动

sudo apt purge nvidia* -y
sudo apt remove nvidia-* -y
sudo rm /etc/apt/sources.list.d/cuda*
sudo apt autoremove -y && sudo apt autoclean -y
sudo rm -rf /usr/local/cuda*

2.2. 更新Ubuntu

sudo apt update
sudo apt upgrade 
# 其他重要软件更新
sudo apt install g++ freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev

2.2. 添加PPA

PPA代表“Personal Package Archive个人软件包存档”，是一种由应用程序开发人员和Linux用户开发和发布的软件存储库，用于存储和分发在官方操作系统存储库中找不到的软件包。

# first get the PPA repository driver
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

2.3. 安装驱动

# 查看推荐驱动版本
ubuntu-drivers devices 

2.4 版本选择

建议的代码是：

# 查看推荐驱动版本
sudo apt install libnvidia-common-515 libnvidia-gl-515 nvidia-driver-515 -y

其他的选项

# 自动安装
sudo ubuntu-drivers autoinstall 
# 选择版本安装
sudo apt install nvidia-driver-535

2.5 重启检查

重启

sudo reboot now # 重启

检查

nvidia-smi # 检查

注意，上图中的 CUDA Version 后面写了一个版本号。该版本号并不是你已经安装了该版本的 CUDA 的意思，而是说此显卡最大支持的CUDA版本号。

2.5 最后设置

sudo wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /"

1. sudo wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin**：从指定的URL下载名为“cuda-ubuntu2204.pin”的文件。该文件可能包含针对Ubuntu 22.04上CUDA存储库的软件包固定信息。

2. sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600：将下载的“cuda-ubuntu2204.pin”文件移动到目录“/etc/apt/preferences.d/”，并将其重命名为“cuda-repository-pin-600”。该文件用于指定CUDA存储库的软件包优先级/偏好设置。

3. sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub**：从指定的URL获取GPG密钥，并将其添加到apt用于软件包验证的受信任密钥列表中。此密钥是用于验证CUDA存储库中软件包的真实性。

4. sudo add-apt-repository “deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /”：将CUDA存储库添加到apt存储库列表中。此命令会在“/etc/apt/sources.list”文件或“/etc/apt/sources.list.d/”下的文件中添加一个新条目，允许apt从该存储库下载和安装与CUDA相关的软件包。