pytorch与cuda版本对应
时间: 2024-01-07 09:04:57 浏览: 110
浅谈pytorch、cuda、python的版本对齐问题
5星 · 资源好评率100%
下表列出了常见的PyTorch版本与对应的CUDA版本:
| PyTorch版本 | 对应的CUDA版本 |
| --- | --- |
| 1.9.0 | 11.1, 11.0, 10.2, 10.1 |
| 1.8.0 | 11.1, 11.0, 10.2, 10.1, 9.2 |
| 1.7.0 | 11.0, 10.2, 10.1, 9.2 |
| 1.6.0 | 10.2, 10.1, 9.2 |
| 1.5.0 | 10.2, 10.1, 9.2 |
| 1.4.0 | 10.1, 10.0, 9.2 |
| 1.3.0 | 10.1, 10.0, 9.2 |
| 1.2.0 | 10.0, 9.2 |
| 1.1.0 | 10.0, 9.2 |
| 1.0.0 | 9.0, 8.0 |
需要注意的是,不同版本的PyTorch可能需要不同版本的CUDA。在安装PyTorch时,需要根据自己的CUDA版本选择合适的PyTorch版本。同时,也需要确保自己的GPU支持该版本的CUDA。
阅读全文