PyTorch 与 TensorFlow 比较

最近学习 PyTorch,参考网上大家的讨论,简单的将 PyTorch 和 TensorFlow 做了一下比较(2020/03/30),难免有疏漏,仅供参考

PyTorch 是后来者,但由于其易用性和灵活性,越来越收到青睐,随着时间的推移,二者的比较应该还会持续变动

特性 PyTorch TensorFlow 备注
易用性 与 python 非常契合,与 numpy 非常类似,易于上手 定义图的时候像是在使用一门新的语言,上手难
API 易于使用 混乱, 冗余
图计算 动态图计算,可以在运行时构建计算图,甚至可以在运行时更改它们,非常灵活 Tensorflow 1.x 静态图构造,因此需要对图形进行编译,然后在执行引擎上执行 Tensorflow 2.0 默认使用动态图,对 1.x 的兼容不太好,原有程序可能需要较大改动
可视化 Visdom,TensorBoard TensorBoard
部署生产 开放式神经网络交换ONNX,解决通用深度学习框架之间的模型迁移部署 直接使用 TensorFlow serving 在 TensorFlow 中部署模型 一开始 pytorch 在部署上没有优势,但随着ONNX生态的发展,逐渐和 TensorFlow 拉平
调试 简单,可以使用标准调试器 ,例如pdb或PyCharm TensorFlow eager 之前需要专门的调试器 tfdbg Tensorflow 2.0 默认使用动态图以后和 PyTorch 一样直接使用标准调试器
性能 不相伯仲
文档 全面 全面

参考文章:

有些文章较早不能代表现在的情况

PyTorch攻势凶猛,程序员正在抛弃TensorFlow?

PyTorch和TensorFlow哪家强:九项对比读懂各自长项短板

PyTorch VS TensorFlow谁最强?标星15000+Transformers的运行结果

Tensorflow eager execution 模式

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • PyTorch VS TensorFlow:细数两者的不同之处 在今年 5 月初召开的 Facebook F8 开...
    Lornatang阅读 20,321评论 11 6
  • 围棋号称人类最复杂的棋类运动,但近两年来,在AlphaGo的冲击下,已经溃不成军。继2016年AlphaGo以4:...
    金字塔下的小蜗牛阅读 1,839评论 3 9
  • 今天,Caffe2和PyTorch突然成了热点词。。。深度计算的几个引擎:Caffe2,PyTorch,Tenso...
    smilegator阅读 1,469评论 0 1
  • 我今天背了古诗元日,宋,王安石,爆竹声中一岁除,春风送暖入屠苏,千门万户曈曈日,总把新桃换旧符,主要是说一年过去了...
    高锦浩阅读 81评论 0 0
  • 姓名 :秦飞 (单位)扬州市方圆建筑工程有限公司 【日精进打卡第46天】 哈尔滨363期 感谢一组 【知~学习】 ...
    fy秦飞阅读 91评论 0 0