安装方式选择 📌
根据您的环境选择合适的安装方法:
1. 使用 Conda 安装 🐍
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
✅ 适用于 Anaconda 用户
🔧 安装前请确保已配置好 CUDA 环境
2. 使用 Pip 安装 📦
pip3 install torch torchvision torchaudio
📦 支持 CPU 和 GPU 版本
💡 建议使用虚拟环境:python -m venv pytorch_env
3. 从源码编译 ⚙️
git clone https://github.com/pytorch/pytorch.git
cd pytorch
git checkout v2.0.1
python setup.py build develop
⚙️ 需要较长时间,适合高级用户
⚠️ 请参考官方文档获取完整编译指南 [/ai_ml_tutorials/install_pytorch_advanced]
验证安装 🧪
安装完成后执行:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
✅ 成功输出版本号和CUDA可用状态
🔧 若遇到问题,请查看 PyTorch安装常见问题
环境配置建议 📚
- 推荐使用 Python 3.8-3.11 版本
- CUDA 11.3 与 PyTorch 1.13 兼容性最佳
- 可通过 PyTorch官方文档 获取最新安装说明
📌 安装完成后,建议继续阅读 PyTorch基础教程 了解入门知识