Jetson nano-安装mxnet

1.准备相应的依赖包
sudo apt update
sudo apt -y install \
                        build-essential \
                        git \
                        graphviz \
                        libatlas-base-dev \
                        libopencv-dev \
                        python-pip

sudo pip install --upgrade \
                        pip \
                        setuptools

sudo 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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值