pytorch1.12.1匹配cuda哪个版本
时间: 2023-08-03 14:04:12 浏览: 298
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
在PyTorch官方文档中,可以看到PyTorch 1.2.0和1.2.1版本的CUDA版本匹配如下:
- CUDA 10.0:PyTorch 1.2.0和1.2.1
- CUDA 9.2:PyTorch 1.2.0和1.2.1
- CUDA 9.0:PyTorch 1.2.0
所以,如果你想使用PyTorch 1.2.1,可以选择CUDA 10.0或CUDA 9.2版本。建议查看PyTorch官方文档中的CUDA版本匹配表来确认。
阅读全文