cuda10.1对应的torch版本
时间: 2023-09-16 08:13:01 浏览: 916
cuda10.1-cuda11.0-cuda11.1各版本windows系统下载
5星 · 资源好评率100%
如果你使用的是 NVIDIA 的显卡,建议使用 CUDA 10.1 对应的 PyTorch 版本是 1.4.0 或更高版本。你可以使用以下命令来安装:
```
pip install torch torchvision -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
其中 `-f` 参数用于指定 PyTorch 的下载源为 CUDA 10.1 版本。
阅读全文