PyTorch 1.7.1 的可以安装cuda10.1的指令吗
时间: 2024-04-03 15:36:47 浏览: 109
CUDA10.1向下兼容10.0,完美运行官方TensorFlow、Pytorch_pytorchcuda10.0,cuda10.0可以升级
5星 · 资源好评率100%
可以安装,PyTorch 1.7.1 官方建议使用 CUDA 10.1 版本,可以通过以下命令安装:
```
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/cu101/torch_stable.html
```
这条命令将会安装 PyTorch 1.7.1、torchvision 0.8.2 和 torchaudio 0.7.2 版本,并使用 CUDA 10.1 进行编译。
需要注意的是,如果你使用 CUDA 10.1 版本,你需要安装相应版本的 cuDNN 库,并且你的 NVIDIA 驱动程序版本需要与 CUDA 10.1 版本匹配。你可以在 NVIDIA 官网上查找相应版本的 cuDNN 库和驱动程序,并进行安装。
阅读全文