Skip to main content
 Web开发网 » 站长学院 » 浏览器插件

TensorFlow、Pytorch、OneFlow,MXNet、MindSpore这些框架谁最好用?

2021年11月03日8760百度已收录

公司常用的框架有tensorflow 和 pytorch 。下面描述一下企业对框架的使用现状。

tensorflow 。tensorflow 版本之间变化有点大,1.15的很多api,在2.0后弃用了,对运行环境版本升级有一定的影响。只能是旧的项目沿用旧版本,新项目用新版本;对于tensorflow 的使用呢,有些人直接用原生的tensorflow api,有些人喜欢用封装后的keras,底层的计算执行庫还是由tensorflow 提供。

pytorch 。有些人对于tensorflow 版本混乱的不满,慢慢转投了pytorch 的怀抱。pytorch的api较为稳定,而且提供的多机多卡分布式训练的api也较为简单易用。

个人觉得这两个框架都挺不错,对于怎么选择的话看个人对框架的用途,熟悉程度,与个人偏好吧。如果仅仅是使用框架,两者区别不是很大。如果是想研究框架本身与机器学习深度学习底层计算算法以及对算法进行优化,pytorch 作为后来者近年来对gpu运行加速上不断的努力前行,更受国内外研究人员的青睐。

最后,算法岗其实不仅仅是研究算法,还要涉及算法落地完整流程,包括数据预处理,算法设计与实现,训练模型,保存模型,部署serving服务等流程,有完整一套产业链,来保证算法相关服务正常运行。从这个角度来讲,个人更偏向api稳定的pytorch。版本之间如果变化太大,不利于系统环境升级以及后续的代码维护。

以上仅为个人拙见,仅供参考,有误之处多多见谅。

评论列表暂无评论
发表评论
微信