1.准备相应的依赖包
sudo apt update
sudo apt -y install \build-essential \git \graphviz \libatlas-base-dev \libopencv-dev \python-pipsudo pip install --upgrade \pip \setuptoolssudo pip install \graphviz==0.8.4 \jupyter \numpy==1.15.2
2.下载mxnet源码
git clone --recursive https://github.com/apache/incubator-mxnet.git mxnet
3.在家目录下的.profile文件添加如下内容
export PATH=/usr/local/cuda/bin:$PATH
export MXNET_HOME=$HOME/mxnet/
export PYTHONPATH=$MXNET_HOME/python:$PYTHONPATH
bash source .profile
4.配置CUDA
nvcc --version #检查cuda版本# 进行如下修改
sudo rm /usr/local/cuda
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
5.安装mxnet
下载地址:
1:https://s3.us-east-2.amazonaws.com/mxnet-public/install/jetson/1.4.0/mxnet-1.4.0-cp36-cp36m-linux_aarch64.whl (python3)
2:https://s3.us-east-2.amazonaws.com/mxnet-public/install/jetson/1.4.0/mxnet-1.4.0-cp27-cp27mu-linux_aarch64.whl (python2)
sudo pip install mxnet-***.whl
6.下载libmxnet.so并放在$MXNET_HOME/lib文件夹下
下载地址:
https://s3.us-east-2.amazonaws.com/mxnet-public/install/jetson/1.4.1/libmxnet.so
cd $MXNET_HOME
git checkout v1.4.x
git submodule update --init --recursive
cd python
sudo pip install -e .
参考链接:
https://mxnet.apache.org/get_started/jetson_setup