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

系统要求

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

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

安装步骤

  1. 打开命令行（Windows：开始菜单 -> 运行；macOS/Linux：Terminal）。
  2. 输入以下命令来安装 PyTorch：
pip install torch torchvision torchaudio
  1. 安装完成后，您可以通过以下命令验证安装：
python -c "import torch; print(torch.__version__)"

这将输出安装的 PyTorch 版本。

高级选项

如果您需要 GPU 加速，请在安装时指定您的 CUDA 版本：

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

这将为您的系统安装支持 CUDA 的 PyTorch。

扩展阅读

要了解更多关于 PyTorch 的信息，请访问我们的PyTorch 教程页面。

PyTorch Logo