pip 下载gpu版本的pytorch
时间: 2023-09-26 17:11:43 浏览: 110
pytorch gpu版
要下载 GPU 版本的 PyTorch,可以使用以下命令:
```python
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
请注意,上面的命令使用的是 CUDA 11.1 版本的 PyTorch。如果您使用的是其他版本的 CUDA,请根据需要更改 `-f` 标志的 URL。
阅读全文