在windows系统中cuda_11.1 python3.9.7 对应的torch,torchvision,torchaudio 哪个版本
时间: 2023-06-19 17:08:59 浏览: 180
torch和torchvision版本树莓派python3.7专用
根据官方文档,Windows系统中 CUDA 11.1 和 Python 3.9.7 对应的 PyTorch 版本为 1.9.0,torchvision 版本为 0.10.0,torchaudio 版本为 0.9.0。你可以通过以下命令安装:
```
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
注意,以上命令假设你使用的是 CUDA 11.1 和 Python 3.9.7,并且已经正确安装了 CUDA 和 cuDNN。如果你使用的是其他 CUDA 版本或 Python 版本,需要根据官方文档进行相应调整。
阅读全文