pytorch版本为1.12.1则需要下载那个cuda版本
时间: 2024-05-30 13:09:15 浏览: 192
1.10.0GPU版本pytorch环境,conda安装支持cuda10.2和11.3 pip安装支持cuda10.2和11.1
PyTorch 1.12.1 支持的 CUDA 版本如下:
- CUDA 10.2
- CUDA 11.1
- CUDA 11.3
你可以根据自己的环境选择下载相应的 CUDA 版本。如果你的显卡支持 CUDA 10.2,建议选择 CUDA 10.2,因为它是 PyTorch 1.12.1 官方推荐的 CUDA 版本。
阅读全文