在 Linux 上,PyTorch 官方支持的版本主要取决于 Python 版本、CUDA 版本以及 Linux 发行版(主要是 glibc 版本)。下面给你一个当前(2024–2025 年)常见支持情况的汇总。
| 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 个稳定版。
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 |
✅ 说明:
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 支持 |
|---|---|
| CPU | ✅ |
| NVIDIA GPU (CUDA) | ✅ |
| AMD GPU (ROCm) | ✅(Ubuntu 为主) |
| Intel GPU | ⚠️(实验性) |
python --version
nvidia-smi
nvcc --version
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。
如果你愿意,可以告诉我:
我可以直接帮你给出最合适的安装命令。