PyTorch 是一个流行的开源机器学习库，用于应用深度学习。以下是如何在 Python 中安装 PyTorch 的步骤。

系统要求

在安装 PyTorch 之前，请确保您的系统满足以下要求：

  • 操作系统：Windows、macOS 或 Linux
  • Python：Python 3.6 或更高版本

安装步骤

  1. 访问 PyTorch 官网：PyTorch 官网
  2. 选择合适的版本：根据您的操作系统和 Python 版本选择合适的 PyTorch 版本。
  3. 使用 pip 安装：在终端或命令提示符中运行以下命令：
pip install torch torchvision torchaudio

这将安装 PyTorch 以及其依赖项。

验证安装

安装完成后，您可以使用以下命令验证 PyTorch 是否已正确安装：

python -c "import torch; print(torch.__version__)"

这将打印出当前安装的 PyTorch 版本。

扩展阅读

要了解更多关于 PyTorch 的信息，请访问以下链接：

PyTorch Logo