mxnet安装(Jetson)

mxnet安装

mxnet是一个跨平台的深度学习框架，今天我们将在板子(Jetson Tk1或Jetson Tx1或者Tegra K1)上安装mxnet,并用到cuda加速。

1.第一步，我们先安装依赖

sudo apt-get update
sudo apt-get install -y build-essential git libatlas-base-dev libopencv-dev

2.第二步

git clone --recursive https://github.com/dmlc/mxnet

注意:

一定要仔细检查，看下载是否成功，如果failed，就得重下。

3.第三步

我们得改一些东西

先进入mxnet里面的make文件夹，更改config.mk

USE_CUDA=1

USE_CUDA_PATH = /usr/local/cuda

这里CUDA是可选的，如果你的电脑不能使用cuda(嵌入式开发板(非Jetson系列)，虚拟机，非英伟达的显卡)可以不用改上面那个文件夹。

4.第四步

如果你是非嵌入式下一步可以到mxnet这个文件夹下make -j4

5.第五步

如果你是嵌入式的你得进行如下更改

cd mshadow/make

用你喜欢的编辑器对mshadow.mk进行更改

先找到-msse3，然后把它给删掉，并在下面加一行

ADD_CFLAGS=-DMSHADOW_USE_SSE=0

然后保存退出。

接着进入dmlc文件夹,修改Makefile文件

把-msse2给去掉，然后在下面一行加上

ADD_CFLAGS=-DMSHADOW_USE_SSE=0

接着保存退出。

6.第六步

然后回到mxnet文件夹里面make -j4就行了

7.第七步

如果上述的步骤没有问题，我们就可以来测试一下。

我们可以用下面这个语句

python example/image-classification/train_mnist.py