PyTorch
你可能会说
学深度学习,TensorFlow 还是 PyTorch?
研究和生产的默认深度学习框架动态图、写法像 Python、调试直观,论文和开源模型几乎都先给 PyTorch;部署可转 TorchScript / ONNX。
也常被叫作深度学习框架torch
什么时候选它
- 训练、微调和复现论文
- 需要灵活调试和自定义层
- 用 Hugging Face 生态(默认支持)
什么时候别选
- 只需要调用现成模型 API
- 边缘部署工具链只支持特定格式
- 团队全是 TF1 存量代码
容易混在一起 · 怎么分
PyTorch≠TensorFlow
PyTorch 研究生态和调试体验更好;TensorFlow 在 TF Serving / TFLite 部署链和部分工业场景仍有存量。
Q面试会问什么
- 问:动态图和静态图的区别?
答:动态图边跑边建、易调试;静态图先定义后执行,利于图优化和部署(TF1 风格)。 - 问:autograd 怎么工作?
答:前向时记录计算图,反向按链式法则求梯度;no_grad 可关闭以省显存。
延伸阅读 · 权威出处