PyTorch 是深度学习领域广泛使用的框架,安装方式多样,以下是常见方法:
1. 使用 pip
安装 📦
✅ 推荐方式:适用于大多数用户
pip install torch torchvision torchaudio
📌 注意:确保已安装 Python 3.7+,若需指定 CUDA 版本,请参考 PyTorch 官方文档
2. 使用 conda
安装 🐍
✅ 适合数据科学环境:
conda install pytorch torchvision torchaudio cudatoolkit -c pytorch
📌 提示:Conda 会自动管理依赖,推荐在 Anaconda 环境中操作
3. 源码编译 🧰
✅ 开发者专属:
git clone https://github.com/pytorch/pytorch.git
cd pytorch
📌 需前置条件:安装 CMake 和 Python 开发工具
4. 其他环境适配 🌍
- Windows:推荐使用 PyTorch Windows 安装包
- macOS:需确保 Homebrew 已安装
- Linux:支持 Ubuntu/Debian、CentOS 等系统
常见问题排查 🛠️
- 安装失败?请检查 PyTorch 安装常见错误
- 需要 GPU 加速?请确认 CUDA 驱动版本与 PyTorch 的兼容性
📌 扩展阅读:PyTorch 入门教程