上QQ阅读APP看书,第一时间看更新
2.1.6 Caffe
Caffe是基于C++编写的深度学习框架,作者是贾杨清,源码开放(具有Licensed BSD)并提供了命令行工具以及Matlab和Python接口。
Caffe一直是深度学习研究者使用的框架,很多研究人员在上面进行开发和优化,因而有了不少沉淀,因此Caffe也是流行的深度学习框架之一。尽管如此,Caffe也存在不支持多机、跨平台、可扩展性差等问题。初学使用Caffe还需要注意下面两个问题。
1)Caffe的安装过程需要大量的依赖库,因此涉及很多安装版本问题,初学者须多加注意。
2)当用户要实现一个新的层,就需要用C++实现它的前向传播和反向传播代码,而如果想要新层运行在GPU上,则需要同时使用CUDA实现这一层的前向传播和反向传播。
Caffe2出自Facebook人工智能实验室与应用机器学习团队,贾杨清仍是主要贡献者之一。Caffe2在工程上做了很多优化,比如运行速度、跨平台、可扩展性等,它可以看作Caffe更细粒度的重构,但在设计上,Caffe2其实和TensorFlow更像。目前Caffe2代码已开源。
在工业界和学术界仍有很多人使用Caffe,而Caffe2的出现给我们提供了更多的选择。