核心内容摘要
海角社区在线：连接你我，点亮生活的无限可能
好的以下是 Python、PyTorch、CUDA 和显卡驱动版本的对应关系整理版本对应关系速查表PyTorch 版本支持的 CUDA 版本支持的 Python 版本最低显卡驱动版本要求PyTorch
2.
3
8,
12.
1
8 -
11≥
515.
4
07 (CUDA
11.
/ ≥
530.
3
02 (CUDA
12.
PyTorch
2.
2
8,
12.
1
8 -
11≥
515.
4
07 (CUDA
11.
/ ≥
530.
3
02 (CUDA
12.
PyTorch
2.
1
8,
12.
1
8 -
11≥
515.
4
07 (CUDA
11.
/ ≥
530.
3
02 (CUDA
12.
PyTorch
2.
0
7,
11.
8
8 -
11≥
495.
2
05 (CUDA
11.
PyTorch
1.
1
6,
11.
7
7 -
10≥
450.
8
02 (CUDA
11.
PyTorch
1.
1
2,
1
3,
11.
6
7 -
10≥
450.
8
02 (CUDA
11.
PyTorch
1.
1
2,
1
3,
11.
5
7 -
10≥
450.
8
02 (CUDA
11.
PyTorch
1.
1
2,
1
3,
11.
5
6 -
9≥
450.
8
02 (CUDA
11.
PyTorch
1.
9
2,
11.
1
6 -
9≥
450.
8
02 (CUDA
11.
PyTorch
1.
8
2,
11.
1
6 -
9≥
450.
8
02 (CUDA
11.
PyTorch
1.
7
2,
1
1,
10.
2
6 -
9≥
440.
3
01 (CUDA
10.
重要说明驱动版本要求NVIDIA 驱动版本需≥ CUDA 版本所需的最低驱动例如 CUDA
1
8 要求驱动 ≥
515.
4
07。
可通过以下命令查看驱动版本nvidia-smiPyTorch 与 CUDA 安装PyTorch 官方已集成 CUDA 依赖安装时无需单独配置 CUDA。
推荐使用以下命令安装# 示例安装 PyTorch
3 CUDA
1
1 pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121兼容性建议Python建议使用较新版本如
8避免兼容性问题。
显卡确保 GPU 支持 CUDA如 NVIDIA Tesla/GeForce RTX 系列。
验证安装import torch print(torch.__version__) # PyTorch 版本 print(torch.cuda.is_available()) # 检查 CUDA 是否可用
常见问题Q如何降级 CUDA 版本A直接安装目标版本的 PyTorch 即可如pip install torch
1.
1
0cu117无需手动降级 CUDA。
Q驱动版本足够高但torch.cuda.is_available()返回 FalseA检查 PyTorch 是否与 CUDA 版本匹配或尝试重装 PyTorch。
整理自 PyTorch 官方文档截至 2024 年 6 月实际请以 PyTorch 官网为准。