在 Ubuntu 上，PyTorch 官方支持以下几种“版本维度”的组合，通常你可以从两个层面理解“支持哪些版本”：
PyTorch 在 Ubuntu 上支持 CPU / CUDA 多种版本，目前（2024–2025）常见支持线：
建议：Ubuntu 20.04 / 22.04 / 24.04 优先使用 PyTorch 2.x
PyTorch 官方 wheel 包 明确支持：
| Ubuntu 版本 | 支持情况 |
|---|---|
| Ubuntu 18.04 | ✅（旧版 PyTorch） |
| Ubuntu 20.04 | ✅（推荐） |
| Ubuntu 22.04 | ✅（推荐） |
| Ubuntu 24.04 | ✅（新 PyTorch） |
PyTorch 在 Ubuntu 上支持的 CUDA 版本（随 PyTorch 版本变化）：
| PyTorch 版本 | 支持 CUDA |
|---|---|
| 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+ | 12.1 / 12.4 |
✅ CPU 版本：所有 Ubuntu 都支持
❌ macOS / Windows 的 CUDA 限制不适用于 Ubuntu
pip install torch torchvision torchaudio
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch
| Python | 支持情况 |
|---|---|
| 3.8 | ✅ |
| 3.9 | ✅ |
| 3.10 | ✅（推荐） |
| 3.11 | ✅ |
| 3.12 | ✅（新 PyTorch） |
你可以告诉我：
我可以给你 精确的安装命令 ✅
