最近想入门深度学习,本来打算直接学PyTorch,但看到很多公司招聘要求TensorFlow,又犹豫了。自己跟着教程跑过几个demo,PyTorch的调试确实直观,但一到部署就懵了,ONNX转换各种报错。TensorFlow 2.x的Keras接口也挺简洁,但版本兼容问题搞到头大。想问问大家实际工作中到底用哪个多?如果目标是明年找算法岗,现在应该主攻哪个框架?还是说两个都得会一点?求过来人指点。
楼主
2天前
PyTorch和TensorFlow学哪个?感觉教程看越多越迷茫了
请 登录 后发表回复
全部回复
共 2 条
2楼
1天前
说个实在的,你要是明年找算法岗,现在纠结这个纯粹是浪费时间。我面过不少人,简历上写精通TF结果连动态图静态图都分不清的太多了。PyTorch现在学术界基本是统治地位,你去看顶会论文的代码,十篇里有八篇是PyTorch,面试官问你最近复现过什么模型,你用TF讲起来自己都别扭。TensorFlow在企业里确实存量多,但很多是历史项目,新起的团队尤其是做大模型相关的,PyTorch占绝对优势。部署那块ONNX报错太正常了,跟框架关系不大,是你对模型导出时的动态轴、算子支持没吃透,换个框架照样踩坑。我的建议是主攻PyTorch,把训练和调试链路跑熟,然后花两周了解一下TF的Keras和SavedModel格式就行,面试问到能说清楚区别就够了。两个都学一点这种话听着稳妥,实际上容易两边都不深,面试官一追问就露馅。
3楼
22分钟前
说实话这个问题年年都有人问,我当年也纠结过一模一样的事。我的真实体验是,PyTorch现在是学术和大部分算法岗的主流,尤其你要是投研究型或者大模型相关的岗位,基本清一色PyTorch,面试也大多让你用torch写。TensorFlow在工业部署和一些传统大厂里存量还很多,但新项目真的越来越少了,招聘写TF很多时候是历史遗留,不一定真让你天天写。部署那块你遇到的ONNX报错太正常了,跟框架关系没那么大,更多是算子支持和动态shape的坑,换成TF你也躲不掉。我的建议是主攻PyTorch,把训练、调试、分布式这些吃透,TF花几天看看Keras能读懂别人代码就行,不用硬啃。真要面试前突击一下TF的Serving和TFLite概念足够了,别为了“都学一点”把两边都学成半吊子。