关于 MXNET

MXNet 是一款设计为效率和灵活性的深度学习框架。它允许你混合符号编程和命令式编程,从而最大限度提高效率和生产力。在其核心是一个动态的依赖调度,它能够自动并行符号和命令的操作。MXNet有一个图形优化层,使得符号执行速度快,内存使用高效。这个库便携,轻量,而且能够扩展到多个 GPU 和多台机器。

MXNet官网: http://mxnet.io/
如何安装 MXNet :http://mxnet.io/get_started/install.html

KVStore API
KVStore API负责多GPU的任务分配和它们之间的数据同步。KVStore的使用首先要创建一个kv空间,这个空间用来在不同gpu不同主机间分享数据,最基本的操作是push和pull,push是把数据放入这个空间,pull是从这个空间取数据。

这个空间内保存的是key-value([int, NDArray]),在push/pull的时候来指定到哪个key。这个是个非常棒的设计,提供了很大的自由度,并且为开发者减少了控制底层数据传输的麻烦。
简要实例: http://mxnet.io/api/perl/kvstore.html?highlight=kvstore

相关论文和深度学习知识: