mxnet安裝

mxnet是一個(gè)跨平臺(tái)的深度學(xué)習(xí)框架，今天我們將在板子(Jetson Tk1或Jetson Tx1或者Tegra K1)上安裝mxnet,并用到cuda加速。

1.第一步，我們先安裝依賴(lài)

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注意: 一定要仔細(xì)檢查，看下載是否成功，如果failed，就得重下。

3.第三步

我們得改一些東西 先進(jìn)入mxnet里面的make文件夾，更改config.mk
USE_CUDA=1 USE_CUDA_PATH = /usr/local/cuda
這里CUDA是可選的，如果你的電腦不能使用cuda(嵌入式開(kāi)發(fā)板(非Jetson系列)，虛擬機(jī)，非英偉達(dá)的顯卡)可以不用改上面那個(gè)文件夾。

4.第四步

如果你是非嵌入式下一步可以到mxnet這個(gè)文件夾下make -j4

5.第五步

如果你是嵌入式的你得進(jìn)行如下更改

cd mshadow/make 用你喜歡的編輯器對(duì)mshadow.mk進(jìn)行更改 先找到-msse3，然后把它給刪掉，并在下面加一行 ADD_CFLAGS=-DMSHADOW_USE_SSE=0 然后保存退出。 接著進(jìn)入dmlc文件夾,修改Makefile文件 把-msse2給去掉，然后在下面一行加上 ADD_CFLAGS=-DMSHADOW_USE_SSE=0
接著保存退出。

6.第六步

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

7.第七步

如果上述的步驟沒(méi)有問(wèn)題，我們就可以來(lái)測(cè)試一下。 我們可以用下面這個(gè)語(yǔ)句 python example/image-classification/train_mnist.py

總結(jié)

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

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