内容目录
PyTorch 兼容性
最开始是跑 PyTorch 的时候遇到警告:
Found GPU0 NVIDIA GeForce GT 710 which is of cuda capability 3.5.
PyTorch no longer supports this GPU because it is too old.
The minimum cuda capability supported by this library is 3.7.
经过网上冲浪发现这个 issue:pytorch/pytorch#30532,简单来说就是官方放弃了对 CUDA 3.5 的支持。
于是乎继续网上冲浪找到解决方案:nelson-liu/pytorch-manylinux-binaries
单独安装 torch==1.12.1+cu113
:
pip uninstall torch==1.12.1+cu113
pip install torch==1.12.1+cu113 -f https://nelsonliu.me/files/pytorch/whl/torch_stable.html
然后又遇到:
No matching distribution found for torch==1.12.1+cu113
打开 https://nelsonliu.me/files/pytorch/whl/torch_stable.html 发现 torch==1.12.1+cu113
预编译的 whl 最高仅支持到 Python 3.10,检查当前用的是 Python 3.11,换用 Python 3.10 解决问题。
题外话
在冲浪的过程中,发现了下表(来源)
CUDA 版本 | 最低计算能力 | 弃用计算能力 | 默认计算能力 | 最高计算能力 |
---|---|---|---|---|
5.5(及以下) | 1.0 | N/A | 1.0 | ? |
6.0 | 1.0 | 1.0 | 1.0 | ? |
6.5 | 1.1 | 1.x | 2.0 | ? |
7.x | 2.0 | N/A | 2.0 | ? |
8.0 | 2.0 | 2.x | 2.0 | 6.2 |
9.x | 3.0 | N/A | 3.0 | 7.0 |
10.x | 3.0 | N/A * | 3.0 | 7.5 |
11.x | 3.5 † | 3.x | 5.2 | 11.0:8.0, 11.1:8.6, 11.8:9.0 |
12.x | 5.0 | N/A | 5.2 | 9.0 |
* 计算能力 3.0 在 10.2 中已弃用。
† CUDA 11.5 仍然“支持” 计算能力 3.5 设备。
而什么显卡支持什么版本的 CUDA 可以在 NVIDIA 官方文档中查到:CUDA GPUs – Compute Capability | NVIDIA Developer。
以及旧版本的 CUDA-Toolkit 可以在这里下载到:CUDA Toolkit Archive | NVIDIA Developer。