pip安装CUDA Version: 11.7的pytorch
时间: 2023-06-14 08:06:23 浏览: 481
升级pip到20.0.2+用本地安装包安装pytorch
可以通过以下命令安装CUDA Version: 11.7的PyTorch:
```
pip install torch==1.10.0+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
这将安装PyTorch 1.10.0版本,其中cu111表示使用CUDA版本为11.1,你可以将其替换为cu117以使用CUDA版本为11.7。注意,你的计算机必须支持CUDA 11.7才能成功安装此版本的PyTorch。
阅读全文