【pytorch】主流深度学习框架介绍

主流深度学习框架介绍

Tensorflow

Google开源，使用数据流图( Data Flow Graph )的形式进行计算，节点代表数学运算，线条代表tensor的交互。
目前 Tensorflow 是全世界使用人数最多、社区最为庞大的一个框架。

Caffe

由加州大学伯克利Phd贾扬清开发，对卷积网络支持很好，只提供C++接口。
很多ImageNet比赛网络用Caffe写的，缺点不够灵活，内存占用高，只提供C++接口等。

Torch

支持大量机器学习算法的科学计算框架，pytorch前身。

PyTorch

由Torch7团队开发，以python为开发语言的深度学习框架，能实现GPU加速和动态神经网络（很多主流框架如Tensorflow不支持）
PyTorch可看做GPU加速支持的numpy，也可看做拥有自动求导功能的深度神经网络。

PyTorch相对Tensorflow的优势

Tensorflow是静态的，想要改变网络结构，需要从头构建。
而pytorch通过反向自动求导技术，能够快速且容易地更改神经网络。
动态神经网络是pytorch最大优势。