Ubuntu PyTorch支持哪些GPU

PyTorch在Ubuntu系统上支持的GPU型号相当广泛，涵盖了从入门级到企业级的各种选择。以下是对支持的GPU型号的详细梳理，以及安装CUDA和PyTorch的推荐步骤，帮助您更好地在Ubuntu系统上配置GPU环境。

支持的GPU型号

PyTorch支持的GPU型号包括：

• 数据中心GPU系列：L系列（如L40、L40S、L20、L4、L2）、H系列（如Nvidia H100、H10O、H800、H800 NVL）、A系列（如Nvidia AX800、A800、A100、A40、A30、A16、A10、A2）、RTX系列（如RTX 5000 Ada Generation、RTX 4500 Ada Generation等）、HGX系列（如HGX H20、HGX H800等）、T系列（如Tesla T4）、V系列（如Tesla V100）、P系列（如Tesla P100、P40、P40 M-class）等。
• 消费级GPU系列：包括GeForce系列的部分型号。

安装CUDA和PyTorch的推荐步骤（以Ubuntu 20.04为例）

1. 安装NVIDIA GPU驱动：

• 添加nvidia驱动仓库并更新系统：

sudo add-apt-repository ppa:graphics-drivers/ppasudo apt update

• 安装必要的依赖包并检测GPU：

sudo apt install -y alsa-utilssudo apt install -y pciutils ubuntu-drivers-common
sudo lspci | grep -i nvidia

• 安装推荐的nvidia驱动并重启系统：

sudo ubuntu-drivers devices
sudo apt install nvidia-driver-版本号
sudo reboot

2. 安装CUDA工具套件：

• 添加nvidia cuda仓库：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-wsl-ubuntu.pinsudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-wsl-ubuntu-11-8-local_11.8.0-1_amd64.debsudo dpkg -i cuda-repo-wsl-ubuntu-11-8-local_11.8.0-1_amd64.debsudo cp /var/cuda-repo-wsl-ubuntu-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt get updatesudo apt get -y install cuda

• 验证CUDA安装成功：

cd /usr/local/cuda-11.8/bin./nvcc -V

3. 安装PyTorch：

• 创建并激活conda环境：

conda create -y -n dl-gpu python=3.8
conda activate dl-gpu

• 安装PyTorch：

conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c conda-forge

请注意，具体的CUDA和PyTorch版本可能会随着新版本的发布而发生变化，建议您在安装时查阅最新的官方文档，以确保兼容性和稳定性。