在 Jetson Nano安装MXNet步骤

  • 1 replies
  • 2007 views
*

sisiy

  • *****
  • 175
    • 查看个人资料
在 Jetson Nano安装MXNet步骤
« 于: 四月 16, 2019, 01:38:25 pm »
1. Allocate swap
程序代码: [选择]
fallocate -l 8G swapfile
sudo chmod 600 swapfile
sudo mkswap swapfile
sudo swapon swapfile

2. Install dependencies
程序代码: [选择]
sudo apt-get update
sudo apt-get install -y git build-essential libatlas-base-dev libopencv-dev graphviz python-pip
sudo pip install --upgrade pip
sudo pip install --upgrade setuptools
sudo pip install numpy==1.15.2
sudo pip install graphviz jupyter

3. Build MXNet on Nano
程序代码: [选择]
git clone https://github.com/apache/incubator-mxnet.git --branch v1.4.x --recursive
cd incubator-mxnet/
程序代码: [选择]
cp make/config.mk .
sed -i 's/USE_CUDA = 0/USE_CUDA = 1/' config.mk
sed -i 's/USE_CUDA_PATH = NONE/USE_CUDA_PATH = \/usr\/local\/cuda/' config.mk
sed -i 's/USE_CUDNN = 0/USE_CUDNN = 1/' config.mk
sed -i '/USE_CUDNN/a CUDA_ARCH := -gencode arch=compute_53,code=sm_53' config.mk
程序代码: [选择]
make -j3
4.Install MXNet
 
程序代码: [选择]
cd python
sudo python setup.py install
cd ..
export MXNET_HOME=$(pwd)
echo "export PYTHONPATH=$MXNET_HOME/python:$PYTHONPATH" >> ~/.bashrc
source ~/.bashrc

*

sisiy

  • *****
  • 175
    • 查看个人资料
Re: 在 Jetson Nano安装MXNet步骤
« 回复 #1 于: 四月 16, 2019, 01:46:15 pm »
 prebuilt file for MXNet on Nano. You can install it with pip directly.

Python2.7
Please download the wheel file https://drive.google.com/open?id=1ot6XtrV9r70wUzM13bpTzrGiJWlfUddV
程序代码: [选择]
sudo apt-get install -y git build-essential libatlas-base-dev libopencv-dev graphviz python-pip
sudo pip install mxnet-1.4.0-cp27-cp27mu-linux_aarch64.whl

Python3.7
Please download the wheel file https://drive.google.com/open?id=1jr-kP1_zlLa9tx-GtdlBV3Nn20qRJgzY
程序代码: [选择]
sudo apt-get install -y git build-essential libatlas-base-dev libopencv-dev graphviz python3-pip
sudo pip install mxnet-1.4.0-cp36-cp36m-linux_aarch64.whl