PyTorch 安装教程

PyTorch 是一个流行的开源机器学习库，用于深度学习和计算。以下是使用 PyTorch 的详细安装步骤。

系统要求

  • Python 3.6 或更高版本
  • 确保你的环境中安装了 pip

安装步骤

  1. 打开终端或命令提示符。
  2. 输入以下命令进行安装：
pip install torch torchvision torchaudio

依赖项

  • PyTorch 需要一些依赖项，根据你的操作系统和 Python 版本，可能需要手动安装它们。

Windows

macOS

  • Xcode Command Line Tools
  • Homebrew (brew install python)

Linux

  • 安装 Python 和 pip (sudo apt-get install python3 python3-pip 或根据你的发行版选择合适的命令)

示例

安装后，你可以使用以下命令验证 PyTorch 是否安装成功：

import torch
print(torch.__version__)

如果一切正常，这将输出 PyTorch 的版本号。

扩展阅读

想要了解更多关于 PyTorch 的信息，请访问我们的 PyTorch 官方文档。

PyTorch Logo