选择适合你的安装方式
根据你的开发环境和需求,选择以下安装方法:
使用 pip
安装
pip install torch torchvision torchaudio
📌 提示:确保已安装Python 3.8+,并根据你的CUDA版本选择对应版本的PyTorch。
使用 conda
安装
conda install pytorch torchvision torchaudio -c pytorch
📌 提示:Anaconda用户可直接通过此命令安装,支持更便捷的环境管理。
从源代码编译安装
适合需要自定义配置的高级用户:
- 安装PyTorch官方依赖
- 运行
setup.py
或使用 PyTorch构建工具
验证安装
打开终端执行:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
✅ 若输出版本号和True
,表示安装成功!
常见问题
- ❓ 安装失败?查看PyTorch安装指南
- ❓ 如何选择CPU/GPU版本?参考PyTorch版本对应表
推荐阅读
- PyTorch基础教程 📘
- 深度学习环境配置指南 🛠️
安装完成后,可前往PyTorch官方文档获取更多API信息 📚