cuda10.2对应的torch
时间: 2023-08-18 13:06:11 浏览: 175
pytorch1.9离线版资源,cuda10.2 +对应的cudnn
5星 · 资源好评率100%
您可以使用以下命令安装CUDA 10.2对应的PyTorch:
```
pip install torch==1.7.1+cu102 torchvision==0.8.2+cu102 torchaudio===0.7.2 -f https://download.pytorch.org/whl/cu102/torch_stable.html
```
这将安装PyTorch 1.7.1,对应CUDA 10.2,同时也安装了与其兼容的torchvision和torchaudio。
阅读全文