Ubuntu系统中PyTorch的GPU支持情况如何

Ubuntu系统中PyTorch支持GPU，需安装NVIDIA GPU驱动、CUDA Toolkit、cuDNN库及对应版本的PyTorch，具体如下：

• 支持的GPU型号：包括NVIDIA GeForce（如RTX 30/20/10系列）、Quadro、Tesla系列等，具体需与CUDA版本兼容。
• 安装步骤：
  1. 安装NVIDIA驱动：通过PPA或官网下载对应型号的驱动并安装。
  2. 安装CUDA Toolkit：从NVIDIA官网下载与驱动兼容的版本，通过.deb包或仓库安装。
  3. 安装cuDNN库：注册NVIDIA账号后下载，解压并复制到CUDA目录。
  4. 安装PyTorch GPU版：通过pip或conda安装，需指定与CUDA版本匹配的PyTorch版本（如pip install torch --extra-index-url https://download.pytorch.org/whl/cu117）。
• 验证方法：运行Python代码import torch; print(torch.cuda.is_available())，若返回True且显示GPU型号，则配置成功。

注：WSL2环境下需在Windows主机安装驱动，Ubuntu子系统仅需安装CUDA Toolkit等组件。