如何使用 pip 安装 PyTorch - 简单指南

要安装 PyTorch，请执行以下步骤：

1. 打开终端或命令提示符（Windows）。

2. 输入以下命令以安装 PyTorch：

pip install torch torchvision

如果您电脑上已经安装了 CUDA，可以使用以下命令安装支持 CUDA 的 PyTorch 版本：

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

注意：根据您的操作系统和 Python 版本，可能需要使用不同的命令。请参考 PyTorch 官方文档，以获取适合您的安装命令。

3. 等待安装完成。完成后，您可以在 Python 代码中导入 PyTorch 并开始使用它。

import torch