深度学习框架的介绍与比较(Caffe, TensorFlow, MXNet, Torch, Theano)

转载 2016年08月28日 18:12:58

在这里,我将会介绍当前比较主流的5种深度学习框架,包括 Caffe, TensorFlow, MXNet, Torch, Theano,并对这些框架进行分析。

首先对这些框架进行总览。

库名称

开发语言

速度

灵活性

文档

适合模型

平台

上手难易

Caffe

c++/cuda

一般

全面

CNN

所有系统

中等

TensorFlow

c++/cuda/Python

中等

中等

CNN/RNN

Linux, OSX

MXNet

c++/cuda

全面

CNN

所有系统

中等

Torch

c/lua/cuda

全面

CNN/RNN

Linux, OSX

中等

Theano

python/c++/cuda

中等

中等

CNN/RNN

Linux, OSX

接下来将对这些框架进行分别介绍。


Caffe

第一个主流的工业级深度学习工具。
它开始于2013年底,由UC Berkely的Yangqing Jia老师编写和维护的具有出色的卷积神经网络实现。在计算机视觉领域Caffe依然是最流行的工具包。
它有很多扩展,但是由于一些遗留的架构问题,不够灵活且对递归网络和语言建模的支持很差。

TensorFlow

Google开源的其第二代深度学习技术——被使用在Google搜索、图像识别以及邮箱的深度学习框架。
是一个理想的RNN(递归神经网络)API和实现,TensorFlow使用了向量运算的符号图方法,使得新网络的指定变得相当容易,支持快速开发。
缺点是速度慢,内存占用较大。(比如相对于Torch)

MXNet

是李沐和陈天奇等各路英雄豪杰打造的开源深度学习框架,是分布式机器学习通用工具包DMLC 的重要组成部分。
它注重灵活性和效率,文档也非常的详细,同时强调提高内存使用的效率,甚至能在智能手机上运行诸如图像识别等任务。

Torch
Facebook力推的深度学习框架,主要开发语言是C和Lua。
有较好的灵活性和速度。
它实现并且优化了基本的计算单元,使用者可以很简单地在此基础上实现自己的算法,不用浪费精力在计算优化上面。核心的计算单元使用C或者cuda做了很好的优化。在此基础之上,使用lua构建了常见的模型。
缺点是接口为lua语言,需要一点时间来学习。

Theano
2008年诞生于蒙特利尔理工学院,主要开发语言是Python。
Theano派生出了大量深度学习Python软件包,最著名的包括BlocksKeras
Theano的最大特点是非常的灵活,适合做学术研究的实验,且对递归网络和语言建模有较好的支持,缺点是速度较慢。

参考
深度学习框架的评估与比较
Caffe、TensorFlow、MXnet三个开源库对比
CVPR 2015深度学习回顾:ConvNet、Caffe、Torch及其他
torch7怎么样?和theano和caffe相比如何?顺便问下实现cnn麻烦吗?
深度学习简析,TensorFlow,Torch,Theano,Mxn

系统学习深度学习(七)--主流深度学习开源框架对比

转自:http://blog.csdn.net/u010167269/article/details/51810613,本文做了相关修改。 参考论文:《Comparative Study of De...

深度学习框架的介绍与比较(Caffe, TensorFlow, MXNet, Torch, Theano)

当前比较主流的5种深度学习框架,包括 Caffe, TensorFlow, MXNet, Torch, Theano,并对这些框架进行分析。 首先对这些框架进行总览。 ...

如何选择深度学习框架 TensorFlow/Torch/Mxnet/Theano

原址 要因人而异。回顾DL库的发展历程,可以看到从当年的caffe,theano到现在的这几种库,每一个时期都有不同的用户群。 caffe,theano 这些库当年都是在大学里开发的,主要目的是为...

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比 | 雷锋网 https://www.leiphone.com/news/201702/T5e31Y2ZpeG1ZtaN...
  • yusisc
  • yusisc
  • 2017年11月21日 11:12
  • 124

TensorFlow和Caffe、MXNet、Keras等深度学习框架的对比

Google 近日发布了 TensorFlow 1.0 候选版,这第一个稳定版将是深度学习框架发展中的里程碑的一步。自 TensorFlow 于 2015 年底正式开源,距今已有一年多,这期间 Ten...

TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比

本文节选自《TensorFlow实战》第二章  | 作者:黄文坚、唐源  | Google近日发布了TensorFlow 1.0候选版,这第一个稳定版将是深度学习框架发展中的里...
  • noztwo
  • noztwo
  • 2017年02月22日 06:45
  • 590

深度学习笔记(一)安装Cuda+Theano+Caffe+Torch

在深度学领域,有好几个开源框架,最著名的有Caffe,Theano和Torch。这三个框架各有所长。 Caffe的用户界面是做得最好的也是最容易上手的,但模块都是固定的,扩展性不佳。而且只能处理有向无...

深度学习框架哪家强?MXNet称霸CNN、RNN和情感分析,TensorFlow仅擅长推断特征提取

深度学习框架哪家强:TensorFlow?Caffe?MXNet?Keras?PyTorch?对于这几大框架在运行各项深度任务时的性能差异如何,各位读者不免会有所好奇。 微软数据科学...

深度学习框架tensorflow,paddle,mxnet的实战性能对比测试

选用了业界主流的三种深度学习框架Paddle,TensorFlow和Mxnet,对它们分别作了研究和实战评估。用于测试的模型包括基于logistic回归模型和LSTM模型。...

深度学习框架哪家强?MXNet称霸CNN、RNN和情感分析,TensorFlow仅擅长推断特征提取

深度学习框架哪家强:TensorFlow?Caffe?MXNet?Keras?PyTorch?对于这几大框架在运行各项深度任务时的性能差异如何,各位读者不免会有所好奇。 微软数据科学...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习框架的介绍与比较(Caffe, TensorFlow, MXNet, Torch, Theano)
举报原因:
原因补充:

(最多只允许输入30个字)