cuda11.4对应的pytorch的下载命令
时间: 2023-04-09 15:04:16 浏览: 116
您可以使用以下命令下载与CUDA 11.4兼容的PyTorch:
pip install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
相关问题
用pip下载cuda11.4对应pytorch
的命令是什么?
您可以使用以下命令下载cuda11.4对应的pytorch:
pip install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
cuda11.4安装Pytorch命令
安装 PyTorch(包括 GPU 版本)需要以下命令:
```
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
```
这个命令会安装 PyTorch 1.9.0,包括 CUDA 11.1 的 GPU 版本,以及 torchvision 和 torchaudio。如果需要安装不同版本的 PyTorch 或者不同的 CUDA 版本,请在 `https://download.pytorch.org/whl/` 中选择对应的版本。