如何安装PyTorch

要在您的系统上安装 PyTorch，请按照以下步骤操作：

1. 确保您已经安装了 Python 和 pip。

2. 在命令行窗口中运行以下命令来安装 PyTorch：

pipinstall torch

3. 如果您希望安装 GPU 版本的 PyTorch，请运行以下命令：

pipinstall torch-gpu

注意：在安装 GPU 版本的 PyTorch 时，您还需要确保您的系统已安装 NVIDIA 的显卡驱动和 CUDA 工具包。

4. 安装完成后，您可以通过在 Python 交互式环境中导入 PyTorch 库来验证安装是否成功：

import torchprint(torch.__version__)

如果没有出现任何错误，并且能够输出 PyTorch 的版本号，则说明安装成功。