在机器学习领域，PyTorch VS TensorFlow哪个更强？

Posted 2021-05-02 AI人工智能科技平台

在机器学习领域，面对各类复杂多变的业务问题，构建灵活易调整的模型是高阶机器学习工程师必备的工作能力。然而，许多工程师还是有一个想法上的误区，以为只要掌握了一种深度学习的框架就能走遍天下了。

事实上，在机器学习领域里没有任何一种框架是能够制霸整个行业的，每位机器学习工程师都必须同时掌握多种框架才能适应业务发展的需要。

 

那么有没有一种框架是相对来说更方便好用的呢？这个问题的答案是肯定的，在这里我就要为你推荐 PyTorch了。

不瞒你说，PyTorch 以它良好的扩展性和超高的实现速度，近年来已赢得了不少工程师的喜爱和赞赏。首先，PyTorch 支持 GPU，这就能够显著提升代码的运行效率。同时，相比 TensorFlow 和 Caffe，PyTorch 拥有反向自动求导技术，让你在调整自定义模型的时候不必从头开始，帮助你节省不少的开发时间。

此外，PyTorch 的代码还比 TensorFlow 的代码更加简洁直观、友好易懂，堪称是非常优质的学习案例，这也能帮助许多工程师更深度地理解机器学习。

 

说到这里，不得不推荐极客时间 PyTorch 的直播小课了，本周三、周四，连续 2 天 20:00-22:00 直播，手把手教你上手 PyTorch。现在还有少量免费学习的名额哦！

 

王然手把手教你上手 PyTorch

众微科技 AI Lab 负责人、阿姆斯特丹大学数学和计量经济学双硕士--王然，将从零开始手把手地带你理解 PyTorch 的基本用法、模型训练过程以及使用PyTorch Lighting 完成复杂逻辑，最后实现一个自定义的神经网络并验证效果。

课程大纲