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

ubuntu

小樊

2025-08-11 22:14:37

栏目: 智能运维

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等组件。

最新问答