温馨提示×

linux pytorch支持哪些版本

小樊
74
2026-10-07 08:46:04
栏目: 智能运维

在 Linux 上,PyTorch 官方支持的版本主要取决于 Python 版本、CUDA 版本以及 Linux 发行版(主要是 glibc 版本)。下面给你一个当前(2024–2025 年)常见支持情况的汇总。


一、PyTorch 版本支持概览(Linux)

✅ 当前主流支持版本

PyTorch 版本 发布时间 Linux 支持情况
2.0.x 2023 ✅ 支持
2.1.x 2023 ✅ 支持
2.2.x 2024 ✅ 支持
2.3.x 2024 ✅ 支持
2.4.x 2024 ✅ 支持(最新稳定)
2.5.x 2025 ✅ 支持(最新稳定)

PyTorch 通常同时支持多个版本,但只维护最近 2–3 个稳定版。


二、Python 版本支持(Linux)

PyTorch 官方 wheel 支持的 Python 版本:

PyTorch 版本 Python 支持
2.0+ 3.8 – 3.11
2.2+ 3.8 – 3.12
2.4+ 3.9 – 3.12
2.5+ 3.9 – 3.12

✅ 说明:

  • Python 3.7 已不再支持
  • Python 3.13 仍在适配中(部分版本可能不稳定)

三、CUDA / GPU 支持(Linux)

PyTorch on Linux 支持 NVIDIA CUDA:

CUDA 版本 PyTorch 支持情况
CUDA 11.7 ✅(旧版本)
CUDA 11.8 ✅(广泛支持)
CUDA 12.1 ✅
CUDA 12.4 ✅(2.4+)
CUDA 12.5+ ⚠️ 部分支持

常见安装方式

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

# CPU only
pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu

四、Linux 发行版支持

✅ 官方明确支持

  • Ubuntu 18.04 / 20.04 / 22.04
  • CentOS 7+
  • Debian 10+

⚠️ 实质要求

  • glibc ≥ 2.17(CentOS 7 是底线)
  • 推荐 glibc ≥ 2.28(Ubuntu 20.04+)

五、CPU / ROCm / 其他后端

后端 Linux 支持
CPU ✅
NVIDIA GPU (CUDA) ✅
AMD GPU (ROCm) ✅(Ubuntu 为主)
Intel GPU ⚠️(实验性)

六、查看你系统支持的版本

查看 Python

python --version

查看 CUDA

nvidia-smi
nvcc --version

查看已安装 PyTorch

python -c "import torch; print(torch.__version__, torch.cuda.is_available())"

七、总结一句话

Linux 上目前推荐 PyTorch 2.2–2.5,Python 3.9–3.12,CUDA 11.8 或 12.x。

如果你愿意,可以告诉我:

  • 你的 Linux 发行版
  • Python 版本
  • 是否有 NVIDIA / AMD GPU

我可以直接帮你给出最合适的安装命令。

0 踩