PyTorch安装指南

安装要求:

  • Python 3.6或更高版本
  • pip包管理器

GPU支持(可选):

  • NVIDIA显卡
  • CUDA支持

安装步骤:

  1. 检查并安装pip
  2. 根据CUDA版本下载并安装CUDA Toolkit
  3. 访问PyTorch官方网站获取相应安装命令

示例安装命令(支持CUDA 11.3):

pip install torch torchvision torchaudio cudatoolkit=11.3
docx 文件大小:17KB