ubuntu16.04 安装Mxnet + lstm + ctc

本文档详细介绍了如何在Ubuntu16.04上安装MXNet库,并确保其支持OpenBLAS、OpenCV、CUDA和cuDNN。首先,通过apt-get安装OpenBLAS和libopencv-dev,然后从GitHub克隆MXNet源码并配置编译选项,特别启用CUDA和cuDNN支持。在安装过程中,可能会遇到问题,如jemalloc的链接错误,解决方法是修改环境变量。接着,安装CTC库warp-ctc,并将MXNet配置为使用该库。最后,清理并重新编译MXNet以应用所有更改。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1、安装OpenBLAS

$sudo apt-get install -y libopenblas-dev

2、安装opencv

$sudo apt-get install -y libopencv-dev

3下载mxnet源码并安装

$git clone --recursize https://github.com/dmlc/mxnet

$cd mxnet

$make  -j $(nproc) USE_OPENCV=1 USE_BLAS=openblas USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda USE_CUDNN=1

或拷贝make下面的config.mk到当前目录下

修改如下内容

然后执行make -j8

4、安装python

$cd python

$sudo python setup.py install

5、测试mxnet python环境是否安装成功

$python

>>>import mxnet

出现问题

解决方法:

重新安装jemalloc并在~/.bashrc文件末尾加上

export LD_PRELOAD="/usr/local/lib/libjemalloc.so"

6、安装CTC

cd ~/

git clone https://github.com/baidu-research/warp-ctc

cd warp-ctc

mkdir build

cd build

cmake ..

make

sudo make install

7、让mxnet支持ctc

cd mxnet

修改mxnet的config.mk文件,对文件中的如下两行进行修改并使其生效

WARPCTC_PATH = $(HOME)/warp-ctc

MXNET_PLUGINS += plugin/warpctc/warpctc.mk

8、重新编译Mxnet

cd mxnet

make clean

$make  -j $(nproc) USE_OPENCV=1 USE_BLAS=openblas USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda USE_CUDNN=1

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值