Questions about installing mxnet and gluoncv on nvidia jetson tx2 #14953
-
|
Beta Was this translation helpful? Give feedback.
Replies: 5 comments
-
Hey, this is the MXNet Label Bot. |
Beta Was this translation helpful? Give feedback.
-
@mxnet-label-bot add [question, build] |
Beta Was this translation helpful? Give feedback.
-
My device is nvidia jetson tx2
sudo apt-get update
sudo apt-get -y install git build-essential libatlas-base-dev libopencv-dev graphviz python-pip
sudo pip install pip --upgrade
sudo pip install setuptools numpy --upgrade
sudo pip install graphviz jupyter
git clone https://github.com/apache/incubator-mxnet.git --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_62,code=sm_62' config.mk
Editing 3rdparty/mshadow/make/mshadow.mk and adding the following after line 122:
MSHADOW_CFLAGS += -DMSHADOW_USE_PASCAL=1
make -j $(nproc)
cd .. # (incubator-mxnet dir)
export MXNET_HOME=$(pwd)
echo "export PYTHONPATH=$MXNET_HOME/python:$PYTHONPATH" >> ~/.bashrc
source ~/.bashrc The main difference between the above steps and the official website tutorial is that: |
Beta Was this translation helpful? Give feedback.
-
@Jessespace Can you try the instructions here: https://github.com/apache/incubator-mxnet/blame/master/docs/install/install-jetson.md |
Beta Was this translation helpful? Give feedback.
-
@Jessespace I've tried installation on Jetson TX2 before and I would definitely recommend using the cross-compilation method. |
Beta Was this translation helpful? Give feedback.
@Jessespace I've tried installation on Jetson TX2 before and I would definitely recommend using the cross-compilation method.