安装方法一览
以下是安装 PyTorch 教程的几种常用方式:
📦 方法一:使用 pip 安装
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
✅ 适用于大多数 Python 环境
🔗 了解更多安装选项
🐍 方法二:使用 conda 安装
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
💡 需要提前配置好 Conda 环境
🖼️
🧰 方法三:从源代码编译
git clone https://github.com/pytorch/pytorch.git
cd pytorch
git checkout release/2.0.0
🔧 需要具备编译环境与依赖项
📌 查看官方编译文档
学习资源推荐
- PyTorch 官方教程(推荐从基础开始学习)
- 深度学习入门课程(适合初学者的系统化指导)
- PyTorch 源码结构解析(进阶开发者可参考)
安装验证
安装完成后,通过以下代码验证是否成功:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
✅ 若输出版本号及 True
表示安装成功
⚠️ 遇到问题请查看 常见错误排查