安装pytorch用什么命令
时间: 2023-12-11 18:58:11 浏览: 64
如何安装pytorch
你可以使用以下命令安装最新版本的 PyTorch:
```python
pip install torch torchvision torchaudio
```
如果你的电脑支持 CUDA,你可以使用以下命令安装支持 CUDA 的 PyTorch:
```python
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
其中,`cu111` 表示 CUDA 11.1 版本,如果你的 CUDA 版本不同,请根据实际情况修改。
阅读全文