安装CUDA 11.8对应的PyTorch版本是1.8.2的torch代码
时间: 2023-12-03 10:44:03 浏览: 543
torch_spline_conv-1.2.2+pt20cu118-cp38-cp38-linux_x86_64.whl.zip
可以使用以下命令安装PyTorch 1.8.2版本:
```bash
pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio==0.8.2 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
这将安装CUDA 11.1和PyTorch 1.8.2。如果你使用的是CUDA 11.8,则可以使用`cu118`代替`cu111`。
阅读全文