当前比较主流的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
TensorFlow
MXNet