PyTorch 是一个深度学习框架,旨在实现简单灵活的实验。PyTorch 是一个基于 Torch 的 Python 开源机器学习库,用于自然语言处理等应用程序。 它主要由 Facebook 的人工智能研究小组开发。Uber 的 “Pyro” 也是使用的这个库。自 2017 年初首次推出,PyTorch 很快成为 AI 研究人员的热门选择并受到推崇。PyTorch 有许多优势,如采用 Python 语言、动态图机制、网络构建灵活以及拥有强大的社群等。由于其灵活、动态的编程环境和用户友好的界面,PyTorch 是快速实验的理想选择。
PyTorch 现在是 GitHub 上增长速度第二快的开源项目,在过去的 12 个月里,贡献者增加了 2.8 倍。而且,去年 12 月在 NeurIPS 大会上,PyTorch 1.0 稳定版终于发布。PyTorch 1.0 增加了一系列强大的新功能,大有赶超深度学习框架老大哥 TensorFlow 之势。
因此,学习 PyTorch 大有裨益!
PyTorch 是一个 Python 包,提供两个高级功能:
- 具有强大的 GPU 加速的张量计算(如 NumPy)
- 包含自动求导系统的的深度神经网络
那么PyTorch还是TensorFlow?
没有好的框架,只有适合的框架, 这里有个简单的对比,大家可以参考,并且技术是发展的,这里的对比也不是绝对的,比如 Tensorflow 在 1.5 版的时候就引入了 Eager Execution 机制实现了动态图,PyTorch 的可视化,window