cuda12.2对应的pytorch版本是什么
时间: 2023-11-19 12:05:03 浏览: 7534
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
对应CUDA 12.2的PyTorch版本是1.10.0。您可以使用以下指令来安装适合CUDA 12.2的PyTorch版本:
```
pip install torch==1.10.0+cu112 -f https://download.pytorch.org/whl/cu112/torch_stable.html
```
请注意,此指令假定您已经安装了CUDA 12.2和适合您的操作系统的驱动程序。如果您还没有安装CUDA 12.2,您需要先安装它。另外,如果您想使用GPU加速,您还需要安装适合CUDA 12.2的cuDNN库。
阅读全文