选择适合你的安装方式 🛠️
1. 使用 pip 安装 📦
pip install torch torchvision torchaudio
📌 提示:确保已安装 Python 3.8+ 和 CUDA Toolkit(如需 GPU 支持)
2. 使用 conda 安装 🐍
conda install pytorch torchvision torchaudio -c pytorch
🔗 本机用户可前往 PyTorch 官方文档 获取更多环境配置建议
3. 从源码编译 🧰
git clone https://github.com/pytorch/pytorch
cd pytorch
git checkout 1.12.1
python setup.py build develop
💡 需要较高系统要求,建议先查看 安装系统依赖
验证安装 🧪
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # GPU 可用性检测
🖼️
常见问题 🤔
- 安装失败?请检查 依赖冲突解决方案
- 需要 CPU 版本?可访问 PyTorch CPU 安装指南