PyTorch和CUDA及Python版本关系
原创大约 1 分钟
PyTorch和CUDA及Python版本关系
近期研究某模型训练时，发现PyTorch版本和CUDA版本不匹配，导致报错。并且提供资料是
Python3.7,如果生产使用Python版本太久,会引起安全升级等一些列问题。
本文主要记录下PyTorch和CUDA及Python版本关系。
参考资料
https://pytorch.org/get-started/previous-versions/
PyTorch 2.0+ 以上版本的关系的表格：
| CUDA 版本 | PyTorch 版本 | torchvision 版本 | torchvideo 版本 | Python 版本 |
|---|---|---|---|---|
| 11.7 | 2.0.0 | 0.15.1 | 0.1.1 | 3.8, 3.9, 3.10 |
| 11.8 | 2.0.1 | 0.15.2 | 0.1.2 | 3.8, 3.9, 3.10, 3.11 |
| 12.1 | 2.1.0 | 0.16.0 | 0.2.0 | 3.9, 3.10, 3.11 |
| 12.2 | 2.1.1 | 0.16.1 | 0.2.1 | 3.9, 3.10, 3.11 |
PyTorch 2.0+ 以下版本的关系的表格：
| CUDA 版本 | PyTorch 版本 | torchvision 版本 | torchvideo 版本 | Python 版本 |
|---|---|---|---|---|
| 10.2 | 1.7.0 | 0.8.1 | 0.2.2 | 3.6, 3.7, 3.8 |
| 10.2 | 1.7.1 | 0.8.2 | 0.2.2 | 3.6, 3.7, 3.8 |
| 11.0 | 1.7.0 | 0.8.1 | 0.2.2 | 3.6, 3.7, 3.8 |
| 11.0 | 1.7.1 | 0.8.2 | 0.2.2 | 3.6, 3.7, 3.8 |
| 10.2 | 1.8.0 | 0.9.0 | 0.4.0 | 3.6, 3.7, 3.8 |
| 10.2 | 1.8.1 | 0.9.1 | 0.4.0 | 3.6, 3.7, 3.8 |
| 11.1 | 1.8.0 | 0.9.0 | 0.4.0 | 3.6, 3.7, 3.8 |
| 11.1 | 1.8.1 | 0.9.1 | 0.4.0 | 3.6, 3.7, 3.8 |
| 10.2 | 1.9.0 | 0.10.0 | 0.4.0 | 3.6, 3.7, 3.8, 3.9 |
| 11.1 | 1.9.0 | 0.10.0 | 0.4.0 | 3.6, 3.7, 3.8, 3.9 |
| 11.1 | 1.9.1 | 0.10.1 | 0.4.0 | 3.6, 3.7, 3.8, 3.9 |
| 10.2 | 1.10.0 | 0.11.0 | 0.4.1 | 3.6, 3.7, 3.8, 3.9 |
| 11.3 | 1.10.0 | 0.11.0 | 0.4.1 | 3.6, 3.7, 3.8, 3.9 |
| 11.3 | 1.10.1 | 0.11.1 | 0.4.2 | 3.6, 3.7, 3.8, 3.9 |
| 10.2 | 1.11.0 | 0.12.0 | 0.4.2 | 3.7, 3.8, 3.9 |
| 11.3 | 1.11.0 | 0.12.0 | 0.4.2 | 3.7, 3.8, 3.9 |
| 11.5 | 1.11.0 | 0.12.0 | 0.4.2 | 3.7, 3.8, 3.9 |
| 10.2 | 1.12.0 | 0.13.0 | 0.5.0 | 3.7, 3.8, 3.9, 3.10 |
| 11.3 | 1.12.0 | 0.13.0 | 0.5.0 | 3.7, 3.8, 3.9, 3.10 |
| 11.6 | 1.12.0 | 0.13.0 | 0.5.0 | 3.7, 3.8, 3.9, 3.10 |
| 11.6 | 1.12.1 | 0.13.1 | 0.5.1 | 3.7, 3.8, 3.9, 3.10 |