温馨提示×

pytorch linux支持哪些版本

小樊
73
2026-10-04 11:34:46
栏目: 智能运维

PyTorch 在 Linux 上支持的版本,主要分为 PyTorch 自身发行版本 和 对应 Linux 系统 / CUDA / Python 版本支持情况 两部分来说明。


一、PyTorch 官方支持的 Linux 发行版

PyTorch 官方 预编译 wheel 包 主要验证并支持以下 Linux 发行版(x86_64 架构):

  • Ubuntu
    • 18.04(早期版本)
    • 20.04(长期支持,最常用)
    • 22.04(当前主流)
  • CentOS / RHEL
    • CentOS 7(已逐步停止支持)
    • CentOS 8 / RHEL 8+
  • Debian
    • Debian 10 / 11 / 12(社区验证较好)

⚠️ 注意:
PyTorch 官方 wheel 依赖 glibc 版本,一般要求 glibc ≥ 2.17(CentOS 7)或更高。


二、PyTorch 支持的 Python 版本(Linux)

以 PyTorch 2.x 为例(2024–2025):

PyTorch 版本 Python 支持
2.0 / 2.1 3.8 – 3.11
2.2 / 2.3 3.8 – 3.11
2.4 / 2.5 3.8 – 3.12

✅ 推荐:

  • Python 3.9 / 3.10(最稳定)
  • Ubuntu 20.04 / 22.04

三、PyTorch 支持的 CUDA 版本(Linux)

PyTorch Linux 版本通常提供:

1️⃣ CPU 版本

  • 无 GPU 要求
  • 适用于所有 Linux

2️⃣ CUDA 支持版本(常见)

PyTorch 版本 CUDA 支持
1.13 11.6 / 11.7
2.0 11.7 / 11.8
2.1 11.8 / 12.1
2.2 11.8 / 12.1
2.3 11.8 / 12.1
2.4 11.8 / 12.1
2.5 11.8 / 12.1 / 12.4

✅ 推荐组合:

  • Ubuntu 22.04 + CUDA 11.8 / 12.1 + PyTorch 2.x

四、架构支持情况

官方 wheel 支持:

  • ✅ x86_64(主流)
  • ✅ aarch64(如 NVIDIA Jetson、ARM 服务器,部分版本)
  • ❌ 非官方架构需自行编译(如 RISC-V)

五、安装方式示例(Linux)

CPU 版本

pip install torch torchvision

CUDA 11.8

pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118

CUDA 12.1

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

六、总结一句话

PyTorch 在 Linux 上支持 Ubuntu / CentOS / Debian 等主流发行版,推荐 Ubuntu 20.04/22.04 + Python 3.9/3.10 + CUDA 11.8/12.1。

如果你有具体环境(如 Ubuntu 版本、显卡型号、Python 版本),我可以帮你推荐最准确的 PyTorch 安装命令。

0 踩