pip 安装pytorch1.4.0 gpu
时间: 2023-04-15 08:01:10 浏览: 670
您可以使用以下命令在GPU上安装PyTorch 1.4.:
```
pip install torch==1.4.+cu100 torchvision==.5.+cu100 -f https://download.pytorch.org/whl/cu100/torch_stable.html
```
其中,`cu100`表示CUDA 10.版本,如果您的CUDA版本不同,请相应更改。如果您没有安装CUDA,则需要安装CPU版本的PyTorch。
阅读全文