Skip to content

Commit 7961566

Browse files
authored
chore: update release build system (#1501)
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 29def86 commit 7961566

File tree

5 files changed

+16
-69
lines changed

5 files changed

+16
-69
lines changed

py/Dockerfile

Lines changed: 0 additions & 11 deletions
This file was deleted.

py/build_whl.sh

Lines changed: 0 additions & 44 deletions
This file was deleted.

py/ci/Dockerfile.ci

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM pytorch/manylinux-builder:cuda11.6
1+
FROM pytorch/manylinux-builder:cuda11.7
22
ARG trt_version
33

44
RUN echo -e "Installing with TensorRT ${trt_version}"

py/ci/build_whl.sh

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,39 +5,43 @@
55
cd /workspace/project/py
66

77
export CXX=g++
8-
export CUDA_HOME=/usr/local/cuda-11.6
8+
export CUDA_HOME=/usr/local/cuda-11.7
99
PROJECT_DIR=/workspace/project
1010

1111
cp -r $CUDA_HOME /usr/local/cuda
1212

1313
build_py37() {
14+
/opt/python/cp37-cp37m/bin/python -m pip install --upgrade pip
1415
/opt/python/cp37-cp37m/bin/python -m pip install -r requirements.txt
1516
/opt/python/cp37-cp37m/bin/python setup.py bdist_wheel --release --ci
16-
#auditwheel repair --plat manylinux2014_x86_64
17+
#LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/python/cp37-cp37m/lib/python3.7/site-packages/torch/lib:/opt/python/cp37-cp37m/lib/python3.7/site-packages/tensorrt/:/usr/local/cuda-11.7/lib64:/usr/local/cuda-11.7/lib64/stubs /opt/python/cp37-cp37m/bin/python -m auditwheel repair --plat manylinux_2_17_x86_64 dist/torch_tensorrt-*-cp37-cp37m-linux_x86_64.whl
1718
}
1819

1920
build_py38() {
21+
/opt/python/cp38-cp38/bin/python -m pip install --upgrade pip
2022
/opt/python/cp38-cp38/bin/python -m pip install -r requirements.txt
2123
/opt/python/cp38-cp38/bin/python setup.py bdist_wheel --release --ci
22-
#auditwheel repair --plat manylinux2014_x86_64
24+
#LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/python/cp38-cp38/lib/python3.8/site-packages/torch/lib:/opt/python/cp38-cp38/lib/python3.8/site-packages/tensorrt/:/usr/local/cuda-11.7/lib64:/usr/local/cuda-11.7/lib64/stubs /opt/python/cp38-cp38/bin/python -m auditwheel repair --plat manylinux_2_17_x86_64 dist/torch_tensorrt-*-cp38-cp38-linux_x86_64.whl
2325
}
2426

2527
build_py39() {
28+
/opt/python/cp39-cp39/bin/python -m pip install --upgrade pip
2629
/opt/python/cp39-cp39/bin/python -m pip install -r requirements.txt
2730
/opt/python/cp39-cp39/bin/python setup.py bdist_wheel --release --ci
28-
#auditwheel repair --plat manylinux2014_x86_64
31+
#LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/python/cp39-cp39/lib/python3.8/site-packages/torch/lib:/opt/python/cp39-cp39/lib/python3.9/site-packages/tensorrt/:/usr/local/cuda-11.7/lib64:/usr/local/cuda-11.7/lib64/stubs /opt/python/cp39-cp39/bin/python -m auditwheel repair --plat manylinux_2_17_x86_64 dist/torch_tensorrt-*-cp39-cp39-linux_x86_64.whl
2932
}
3033

3134
build_py310() {
35+
/opt/python/cp310-cp310/bin/python -m pip install --upgrade pip
3236
/opt/python/cp310-cp310/bin/python -m pip install -r requirements.txt
3337
/opt/python/cp310-cp310/bin/python setup.py bdist_wheel --release --ci
34-
#auditwheel repair --plat manylinux2014_x86_64
38+
#LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/python/cp310-cp310/lib/python3.8/site-packages/torch/lib:/opt/python/cp310-cp310/lib/python3.8/site-packages/tensorrt/:/usr/local/cuda-11.7/lib64:/usr/local/cuda-11.7/lib64/stubs /opt/python/cp38-cp38/bin/python -m auditwheel repair --plat manylinux_2_17_x86_64 dist/torch_tensorrt-*-cp310-cp310-linux_x86_64.whl
3539
}
3640

3741
#build_py311() {
3842
# /opt/python/cp311-cp311/bin/python -m pip install -r requirements.txt
3943
# /opt/python/cp311-cp311/bin/python setup.py bdist_wheel --release --ci
40-
# #auditwheel repair --plat manylinux2014_x86_64
44+
# #auditwheel repair --plat manylinux_2_17_x86_64
4145
#}
4246

4347
build_libtorchtrt() {

py/setup.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,10 @@
2828

2929
CI_RELEASE = False
3030

31-
__version__ = "1.3.0a0"
32-
__cuda_version__ = "11.6"
33-
__cudnn_version__ = "8.4"
34-
__tensorrt_version__ = "8.4"
31+
__version__ = "1.3.0"
32+
__cuda_version__ = "11.7"
33+
__cudnn_version__ = "8.5"
34+
__tensorrt_version__ = "8.5"
3535

3636

3737
def get_git_revision_short_hash() -> str:
@@ -375,9 +375,7 @@ def run(self):
375375
long_description_content_type="text/markdown",
376376
long_description=long_description,
377377
ext_modules=ext_modules,
378-
install_requires=[
379-
"torch>=1.13.0.dev0,<1.14.0",
380-
],
378+
install_requires=["torch>=1.13.0,<1.14.0", "tensorrt>=8.5.1.7,<8.6.0"],
381379
setup_requires=[],
382380
cmdclass={
383381
"install": InstallCommand,

0 commit comments

Comments
 (0)