cuda对应的torch和python版本
时间: 2024-05-31 11:10:35 浏览: 266
下表是CUDA和PyTorch版本之间的对应关系:
| CUDA版本 | PyTorch 1.7 | PyTorch 1.6 | PyTorch 1.5 | PyTorch 1.4 | PyTorch 1.3 |
| -------- | ----------- | ----------- | ----------- | ----------- | ----------- |
| 11.2 | 1.7.1 | 1.6.0 | 1.5.0 | 1.4.0 | Not supported |
| 11.1 | 1.7.0 | 1.6.0 | 1.5.0 | 1.4.0 | Not supported |
| 11.0 | 1.7.0 | 1.6.0 | 1.5.0 | 1.4.0 | Not supported |
| 10.2 | 1.7.0 | 1.6.0 | 1.5.0 | 1.4.0 | 1.3.1 |
| 10.1 | 1.7.0 | 1.6.0 | 1.5.0 | 1.4.0 | 1.3.1 |
| 10.0 | 1.7.0 | 1.6.0 | 1.5.0 | 1.4.0 | 1.3.1 |
此外,Python版本与PyTorch版本之间的对应关系如下:
| Python版本 | PyTorch 1.7 | PyTorch 1.6 | PyTorch 1.5 | PyTorch 1.4 | PyTorch 1.3 |
| ---------- | ----------- | ----------- | ----------- | ----------- | ----------- |
| 3.6 | ✓ | ✓ | ✓ | ✓ | ✓ |
| 3.7 | ✓ | ✓ | ✓ | ✓ | ✓ |
| 3.8 | ✓ | ✓ | ✓ | ✓ | |
| 3.9 | ✓ | | | | |
其中 ✓ 表示支持该组合。
阅读全文