Skip to content

Commit 61741a4

Browse files
authored
Update BUILD_WHEEL.md
fix small typos & ditch cu90
1 parent dd4042c commit 61741a4

File tree

1 file changed

+14
-6
lines changed

1 file changed

+14
-6
lines changed

doc/BUILD_WHEEL.md

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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
4749
git 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
```
6062
cd /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
```
6872
cd /work/mxnet/tools/pip
6973
ln -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

7583
The built wheel file is in `dist/`

0 commit comments

Comments
 (0)