Ubuntu14.04 安装mxnet的一些注意事项

2016-12-24  本文已影响0人  麦兜胖胖次

记录一下安装mxnet...使用的服务器上已经有cuda和cudaCNN还有hadoop了,没有装cuda和cudnn的可以看我caffe的安装博客。从github上clone下来mxnet,需要改make下的config.mk。注意事项:

  1. 在mxnet的根目录下make。一定要记得,直接make,直接make,直接make!!!!!! cmake是准备给windows的!!!完了你覆盖了原本的makefile的话就gg了。。。可以make -j#() 根据自己的配置决定。

  2. 改config.mk的时候,主要按照自己的pc配置改USE_CUDA, USE_CUDACNN, 同时我还加了:

# the additional link flags you want to add
ADD_LDFLAGS = -L/home/shixin/local/lib -L/home/shixin/local/lib64
# the additional compile flags you want to add
ADD_CFLAGS = -I/home/shixin/local/include -I/home/shixin/include
  1. 重新make的话记得把build文件夹删掉...或者用别人拷给你的mxnet的话一定要删掉build文件夹,不然会提示路径里面有文件找不到什么的。

  2. 如果遇到以下错误:

Makefile:23: mshadow/make/mshadow.mk: No such file or directory
Makefile:24: dmlc-core/make/dmlc.mk: No such file or directory
Makefile:86: ps-lite/make/ps.mk: No such file or directory

是git clone的时候需要git clone --recursive <url>,因为mxnet依赖于DMLC通用工具包http://dmlc.ml/,--recursive参数可以自动加载mshadow等依赖。

  1. 我使用的是python接口。注意:执行任何demo都需要import find_mxnet,当然前提要保证执行文件夹下有find_mxnet.py:
try:
    import mxnet as mx
except ImportError:
    import os, sys
    curr_path = os.path.abspath(os.path.dirname(__file__))
    sys.path.append(os.path.join(curr_path, ".."))
    import mxnet as mx

sys.path.append添加的路径必须是有安装mxnet的时候python/mxnet文件夹的。不使用find_mxnet.py的话,需要直接把python/mxnet文件夹放到执行的目录下。

  1. 报错找不到libmxnet.so?解决方案:把安装目录/lib下的libmxnet.so复制到python/mxnet中。
上一篇下一篇

猜你喜欢

热点阅读