cuda11.5对应的pytorch-gpu
时间: 2023-07-17 15:57:41 浏览: 253
torchvision-0.12.0+cu115-cp310-cp310-linux_x86_64.whl.zip
对应CUDA 11.5的PyTorch-GPU版本是PyTorch 1.10.0以上的版本。你可以通过以下方式安装:
```shell
pip install torch==1.10.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html
```
这将安装PyTorch 1.10.0并与CUDA 11.5兼容的GPU版本。记得确保已经正确安装了CUDA 11.5和对应的GPU驱动程序。
阅读全文