Skip to content

Commit fa1c778

Browse files
committed
Merge branch 'build-cleanup' into 'release/ngc/21.11'
Cleaned up dist_build.sh See merge request adlsa/TRTorch!4
2 parents c7328d0 + ad4ad92 commit fa1c778

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

docker/dist-build.sh

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,16 @@
11
#!/bin/bash
22

3-
mkdir -p dist
3+
TOP_DIR=$(cd $(dirname $0); pwd)/..
44

5-
cd py && MAX_JOBS=1 LANG=en_US.UTF-8 LANGUAGE=en_US:en LC_ALL=en_US.UTF-8 python3 setup.py bdist_wheel --use-cxx11-abi
6-
7-
cd ..
8-
9-
cp py/dist/* dist/
5+
cd ${TOP_DIR} \
6+
&& mkdir -p dist && cd py \
7+
&& MAX_JOBS=1 LANG=en_US.UTF-8 LANGUAGE=en_US:en LC_ALL=en_US.UTF-8 \
8+
python3 setup.py bdist_wheel --use-cxx11-abi $* || exit 1
109

1110
pip3 install ipywidgets --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org
1211
jupyter nbextension enable --py widgetsnbextension
13-
1412
pip3 install timm
1513

1614
# test install
17-
mkdir -p /opt/torch_tensorrt && pip3 uninstall -y torch_tensorrt && pip3 install dist/*.whl
15+
pip3 uninstall -y torch_tensorrt && pip3 install ${TOP_DIR}/py/dist/*.whl
1816

0 commit comments

Comments
 (0)