PyTorch 是一个开源的机器学习库,广泛用于深度学习和计算机视觉。以下是安装 PyTorch 的步骤:

系统要求

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

安装步骤

  1. 下载安装包:访问 PyTorch 官网,选择合适的安装包下载。

  2. 安装 Python 包管理器:如果您的系统中还没有 Python 包管理器(如 pip),请先安装它。

  3. 使用 pip 安装 PyTorch:打开命令行,执行以下命令:

    pip install torch torchvision torchaudio
    
  4. 验证安装:安装完成后,打开 Python 命令行,输入以下代码,确认 PyTorch 已正确安装:

    import torch
    print(torch.__version__)
    

    如果显示 PyTorch 版本号,说明安装成功。

高级安装选项

  • CUDA 支持:如果您想要使用 CUDA 进行深度学习,需要在安装时选择包含 CUDA 支持的安装包。
  • 使用 conda 安装:可以使用 conda 来安装 PyTorch,它是一个 Python 包管理器和环境管理器。

资源

希望这份指南能帮助您顺利安装 PyTorch。如果您遇到任何问题,可以访问 PyTorch 社区论坛寻求帮助。

PyTorch 安装指南