以下是运行 PyTorch 所需的系统要求和推荐配置。

系统要求

  • 操作系统:Linux, macOS, 或 Windows。
  • Python 版本:Python 3.6, 3.7, 或 3.8。
  • CUDA:如果您想要使用 PyTorch 的 GPU 功能,需要安装 CUDA。

推荐配置

  • CPU:至少双核处理器。
  • GPU:NVIDIA GPU,至少 CUDA 9.0 支持。
  • 内存:至少 8GB 内存。

安装 PyTorch

要安装 PyTorch,请访问我们的 安装指南

注意事项

  • 确保您的 Python 环境中没有其他版本的 PyTorch,以避免版本冲突。

PyTorch Logo