@@ -44,9 +44,11 @@ wget https://github.com/Kitware/CMake/releases/download/v3.15.2/cmake-3.15.2.tar
4444 ./configure && make install -j4 && cd .. && \
4545 rm -r cmake-3.15.2 cmake-3.15.2.tar.gz
4646
47+ # change the url to your repo link if you are doing PR
48+ export SIMPLEDET_URL=https://github.com/tusimple/simpledet
4749git clone --recursive --depth=1 https://github.com/apache/incubator-mxnet /work/mxnet && \
4850 cd /work/mxnet && \
49- git clone https://github.com/Tusimple/simpledet /work/simpledet && \
51+ git clone $SIMPLEDET_URL /work/simpledet && \
5052 cp -r /work/simpledet/operator_cxx/* /work/mxnet/src/operator && \
5153 git clone https://github.com/RogerChern/cocoapi /work/cocoapi && \
5254 mkdir -p src/coco_api && \
@@ -59,17 +61,23 @@ git clone --recursive --depth=1 https://github.com/apache/incubator-mxnet /work/
5961```
6062cd /work/mxnet
6163# remove sm_30
62- sed -i 's/KNOWN_CUDA_ARCHS :=\+\n/KNOW_CUDA_ARCHS := 35 50 60 70\n/' Makefile
63- tools/staticbuild/build.sh cu90/cu100/cu101 pip
64+ sed -i 's/KNOWN_CUDA_ARCHS :=.*/KNOWN_CUDA_ARCHS := 35 50 60 70/' Makefile
65+ # change build config according to the target CUDA version
66+ tools/staticbuild/build.sh cu100 pip
67+ # tools/staticbuild/build.sh cu101 pip
6468```
6569
6670### Package wheel
6771```
6872cd /work/mxnet/tools/pip
6973ln -s /work/mxnet mxnet-build
70- # change the path according your CUDA version
71- LD_LIBRARY_PATH=/work/mxnet/staticdeps/usr/local/cuda-10.1/lib64:/work/mxnet/staticdeps/usr/lib/x86_64-linux-gnu:/work/mxnet/staticdeps/usr/lib/nvidia-418
72- mxnet_variant=CU101 python3 setup.py bdist_wheel
74+
75+ # change the path according to the target CUDA version
76+ LD_LIBRARY_PATH=/work/mxnet/staticdeps/usr/local/cuda-10.0/lib64:/work/mxnet/staticdeps/usr/lib/x86_64-linux-gnu:/work/mxnet/staticdeps/usr/lib/nvidia-410
77+ # LD_LIBRARY_PATH=/work/mxnet/staticdeps/usr/local/cuda-10.1/lib64:/work/mxnet/staticdeps/usr/lib/x86_64-linux-gnu:/work/mxnet/staticdeps/usr/lib/nvidia-418
78+ export LD_LIBRARY_PATH
79+ mxnet_variant=CU100 python3 setup.py bdist_wheel
80+ # mxnet_variant=CU101 python3 setup.py bdist_wheel
7381```
7482
7583The built wheel file is in ` dist/ `
0 commit comments