PyTorch 安装指南 🚀

1. 环境准备 🛠️

2. 安装方法 📦

方法一：使用 pip 安装 📡

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

⚠️ 若需 CPU 版本，请将 cu118 替换为 cpu

方法二：使用 Conda 安装 🐌

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

方法三：从源码编译 🔧

git clone https://github.com/pytorch/pytorch.git
cd pytorch
git checkout release/2.3.0

📌 编译过程可能耗时较长，建议参考 官方文档 获取详细说明

3. 验证安装 ✅

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 若为 True 表示成功启用 GPU

🎉 若输出版本号且支持 CUDA，则安装成功！

4. 常见问题 ❓

  • 安装失败？尝试更新 pip：python -m pip install --upgrade pip
  • 版本冲突？使用 pip show torch 检查当前版本
  • CUDA 不可用？请参考 CUDA 安装指南 配置环境

5. 扩展学习 📚

pytorch_installation