PyTorch 是深度学习领域广泛使用的框架,安装方式多样,以下是常见方法:

1. 使用 pip 安装 📦

推荐方式:适用于大多数用户

pip install torch torchvision torchaudio

📌 注意:确保已安装 Python 3.7+,若需指定 CUDA 版本,请参考 PyTorch 官方文档

pip_installation

2. 使用 conda 安装 🐍

适合数据科学环境

conda install pytorch torchvision torchaudio cudatoolkit -c pytorch

📌 提示:Conda 会自动管理依赖,推荐在 Anaconda 环境中操作

conda_installation

3. 源码编译 🧰

开发者专属

git clone https://github.com/pytorch/pytorch.git
cd pytorch

📌 需前置条件:安装 CMakePython 开发工具

source_code_compilation

4. 其他环境适配 🌍

linux_installation

常见问题排查 🛠️

📌 扩展阅读PyTorch 入门教程