🚀 PyTorch 在 Windows 系统上的安装指南

安装方法概览

1. 通过 pip 安装

   • 打开命令提示符（CMD）或 PowerShell
   • 运行命令：

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

   • 📌 注意：需提前安装 Python 环境
   

2. 通过 Conda 安装

   • 打开 Anaconda Prompt
   • 执行命令：

     conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch
     

   • ✅ Conda 会自动管理依赖库，适合科学计算场景
   

验证安装

• 在 Python 环境中运行以下代码：

  import torch
  print(torch.__version__)
  print(torch.cuda.is_available())
  

• 🧪 若输出版本号且显示 True，则安装成功

常见问题

• ❓ 安装失败：检查网络是否通畅，或尝试更换镜像源
• ❓ CUDA 不可用：确保已安装 NVIDIA 显卡驱动及 CUDA 工具包
• 📘 扩展阅读：PyTorch 官方文档 提供更详细的配置说明