选择适合你的安装方式 🛠️

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 可用性检测

🖼️

PyTorch_安装验证

常见问题 🤔

PyTorch_安装流程图