mxnet

作者: 南风无影 | 来源:发表于2017-12-18 18:25 被阅读88次

mxnet是亚马逊的深度学习平台,优点是快,简洁

李沐
caffe是很优秀的dl平台。影响了后面很多相关框架。
cxxnet借鉴了很多caffe的思想。相比之下,cxxnet在实现上更加干净,例如依赖很少,通过mshadow的模板化使得gpu和cpu代码只用写一份,分布式接口也很干净。
mxnet是cxxnet的下一代,目前实现了cxxnet所有功能,但借鉴了minerva/torch7/theano,加入更多新的功能。

  1. ndarray编程接口,类似matlab/numpy.ndarray/torch.tensor。独有优势在于通过背后的engine可以在性能上和内存使用上更优
  2. symbolic接口。这个可以使得快速构建一个神经网络,和自动求导。
  3. 更多binding 目前支持比较好的是python,马上会有julia和R
  4. 更加方便的多卡和多机运行
  5. 性能上更优。目前mxnet比cxxnet快40%,而且gpu内存使用少了一半。
    目前mxnet还在快速发展中。这个月的主要方向有三,更多的binding,更好的文档,和更多的应用(language model、语音,机器翻译,视频)。地址在 dmlc/mxnet · GitHub 欢迎使用

安装方法

git clone --recursive https://github.com/dmlc/mxnet
mxnet/目录里找到mxnet/make/子目录,把该目录下的config.mk复制到mxnet/目录,用文本编辑器打开,找到并修改以下两行:
USE_CUDA = 1
USE_CUDA_PATH = /usr/local/cuda
make -j32

这样还是会找不到mxnet模块,执行下列语句....
// ~表示我的mxnet安装在home下面
export PYTHONPATH=~/mxnet/python
cp -r ~/mxnet/python/mxnet .
cp ~/mxnet/lib/libmxnet.so mxnet/

相关文章

网友评论

      本文标题:mxnet

      本文链接:https://www.haomeiwen.com/subject/vhgmwxtx.html