pip 安装 PyTorch
时间: 2023-10-27 15:06:13 浏览: 106
如何安装pytorch
要使用 pip 安装 PyTorch,请执行以下步骤:
1. 打开命令行或终端窗口(在 Windows 上,可以按 Win+R,然后输入 cmd 并按回车键打开命令行窗口)。
2. 输入以下命令并按回车键:
```python
pip install torch
```
3. 如果你的系统有 GPU 并且你想使用 GPU 进行训练,请输入以下命令并按回车键:
```python
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
这将安装 PyTorch、TorchVision 和 TorchAudio,并将它们配置为使用 CUDA 11.1(如果你的系统上安装了适当版本的 CUDA 驱动程序)。
4. 等待安装完成即可。
阅读全文