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(嵌入式開發(fā)板(非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

總結(jié)

以上是生活随笔為你收集整理的mxnet安装(Jetson)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。