安装方式概览

PyTorch 提供多种安装方式,Windows 用户推荐使用以下方法:

✅ 使用 pip 安装(推荐)

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

请根据你的 CUDA 版本选择对应链接(如 cu117cu119
📌 若需查看完整版本兼容性表,请访问 /pytorch_versions

⚙️ 从源码编译(高级用户)

  1. 安装 Visual Studio
  2. 安装 CMake
  3. 执行 pip install --upgrade pytorch(会自动触发编译)

验证安装

打开终端运行:

python -c "import torch; print(torch.__version__)"

若输出版本号则表示成功 ✅
💡 需要帮助调试环境?查看 /common_issues 获取解决方案

常见问题

pytorch_install

扩展学习

📌 注意:安装前请确认已安装 Python 并配置环境变量