Ubuntu22.04安装配置PyTorch 2.2.0以上版本深度学习环境

1、安装合适的NVIDIA驱动，在终端运行以下命令：

ubuntu-drivers devices  ##  获取显卡和驱动信息

以我本人的NVIDIA RTX 3090显卡为例，显示结果如下：

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00002204sv00001043sd000087B3bc03sc00i00
vendor   : NVIDIA Corporation
model    : GA102 [GeForce RTX 3090]
driver   : nvidia-driver-470 - distro non-free
driver   : nvidia-driver-535-server - distro non-free
driver   : nvidia-driver-545-open - distro non-free
driver   : nvidia-driver-535-open - distro non-free
driver   : nvidia-driver-535-server-open - distro non-free
driver   : nvidia-driver-545 - distro non-free
driver   : nvidia-driver-470-server - distro non-free
driver   : nvidia-driver-535 - distro non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin

2、选择recommended对应的驱动版本（倒数第2行）进行安装：

sudo apt-get install nvidia-driver-535

然后在终端输入nvidia-smi，若显示显卡信息则表示安装成功。

3、PyTorch 2.2.0以上的版本不需要自己单独安装CUDA和cuDNN，直接根据官方的安装命令选择不同的版本进行安装即可：

pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121

pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121

pip install torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 --index-url https://download.pytorch.org/whl/cu121

pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121