cuda11.0对应的pytorch 安装 的命令行
时间: 2023-09-20 12:07:18 浏览: 135
win10+cuda10.0+pytorch安装
5星 · 资源好评率100%
如果您要安装 CUDA 11.0 对应的 PyTorch,可以使用以下命令行进行安装:
```
pip install torch==1.9.0+cu110 torchvision==0.10.0+cu110 torchaudio==0.9.0 -f https://download.pytorch.org/whl/cu110/torch_stable.html
```
这将安装 PyTorch 1.9.0、TorchVision 0.10.0 和 TorchAudio 0.9.0,其中 `cu110` 表示 CUDA 11.0 版本。注意,您需要先安装 CUDA 11.0 和对应的 CuDNN 版本。
阅读全文