Meta 开源的深度学习框架,动态图 API 与完整生态,是 AI 领域应用最广的框架。
Project story
PyTorch 是深度学习领域应用最广的框架:动态计算图让调试与实验直觉化,Tensor API 简洁,从科研原型到生产部署有完整链路,主流开源模型(Llama、Qwen、Stable Diffusion 等)几乎都以它为基础。
提供 torchvision、torchaudio、torchtext 等扩展库,支持 CUDA 加速、分布式训练与移动端导出(TorchScript)。
适合场景:模型开发与训练、AI 科研、生产推理、学习深度学习。
上手方式:pip install torch,按教程定义网络并训练即可。