cuda10.1对应的torch版本号
时间: 2023-10-22 21:03:33 浏览: 83
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
对应的 PyTorch 版本号应该是 1.4.0,可以通过以下命令安装:
```shell
pip install torch torchvision -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
其中,`cu101` 表示 CUDA 10.1 版本。如果你的 CUDA 版本不同,需要相应地修改后面的文件路径。
阅读全文