选型

PyTorch

你可能会说

学深度学习,TensorFlow 还是 PyTorch?

研究和生产的默认深度学习框架动态图、写法像 Python、调试直观,论文和开源模型几乎都先给 PyTorch;部署可转 TorchScript / ONNX。
也常被叫作深度学习框架torch
先问一句要训练 / 微调模型?
默认PyTorch→擅长动态图易调试→别用只调 API
成熟度采用

什么时候选它

  • 训练、微调和复现论文
  • 需要灵活调试和自定义层
  • 用 Hugging Face 生态(默认支持)

什么时候别选

  • 只需要调用现成模型 API
  • 边缘部署工具链只支持特定格式
  • 团队全是 TF1 存量代码
容易混在一起 · 怎么分
PyTorch≠TensorFlow

PyTorch 研究生态和调试体验更好;TensorFlow 在 TF Serving / TFLite 部署链和部分工业场景仍有存量。

Q面试会问什么

  • 问:动态图和静态图的区别?
    答:动态图边跑边建、易调试;静态图先定义后执行,利于图优化和部署(TF1 风格)。
  • 问:autograd 怎么工作?
    答:前向时记录计算图,反向按链式法则求梯度;no_grad 可关闭以省显存。
延伸阅读 · 权威出处