nvidia-smi输出示例:
CUDA Version: 12.2⚠️ 注意:
nvidia-smi 显示的是 驱动支持的最高 CUDA 版本,不是你当前安装的 CUDA Toolkit 版本。nvcc --version或
/usr/local/cuda/bin/nvcc --versioncat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2或(旧版本):
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2| 驱动版本 | 支持的最高 CUDA |
|---|---|
| ≥ 525 | CUDA 12.x |
| ≥ 470 | CUDA 11.4–11.8 |
| ≥ 450 | CUDA 11.0–11.3 |
✅ 规则:
新驱动支持旧 CUDA,但旧驱动不支持新 CUDA
例如:
| GPU 架构 | 最低 CUDA |
|---|---|
| Turing(RTX 20 系) | CUDA 10+ |
| Ampere(RTX 30 系) | CUDA 11+ |
| Ada(RTX 40 系) | CUDA 11.8+ |
| Hopper(H100) | CUDA 12+ |
| PyTorch | CUDA | cuDNN |
|---|---|---|
| 2.1 | 11.8 / 12.1 | 8.7+ |
| 2.0 | 11.7 / 11.8 | 8.5+ |
| 1.13 | 11.6 / 11.7 | 8.3+ |
✅ 查看 PyTorch 实际使用的 CUDA:
import torch
print(torch.version.cuda)
print(torch.cuda.is_available())| TensorFlow | CUDA | cuDNN |
|---|---|---|
| 2.15 | 12.1 | 8.9 |
| 2.12 | 11.8 | 8.6 |
| 2.10 | 11.2 | 8.1 |
1. 先看 GPU 型号
↓
2. 查驱动支持的最高 CUDA
↓
3. 看框架要求的 CUDA
↓
4. 安装对应 CUDA Toolkit + cuDNN✅ NVIDIA CUDA Toolkit 发布说明
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
✅ PyTorch 安装说明
https://pytorch.org/get-started/previous-versions/
✅ TensorFlow GPU 支持
❌ nvidia-smi ≠ 当前 CUDA Toolkit
❌ CUDA 12 不一定能跑 PyTorch 1.x
❌ 驱动太旧会导致 CUDA 不可用
如果你愿意,可以直接把下面信息发我,我可以 直接告诉你该装哪个版本:
nvidia-smi
nvcc --version
python - << 'EOF'
import torch
print(torch.__version__)
print(torch.version.cuda)
EOF或者告诉我:
我可以给你 一条命令直接装对版本 ✅