Merge pull request #1042 from NVIDIA/python_release_system_update

chore: update the release system for python

Author: narendasan

py/Dockerfile

@@ -2,10 +2,9 @@ FROM pytorch/manylinux-cuda113
 
 RUN yum install -y ninja-build
 
-RUN wget --no-check-certificate https://copr.fedorainfracloud.org/coprs/vbatts/bazel/repo/epel-7/vbatts-bazel-epel-7.repo \
-    && mv vbatts-bazel-epel-7.repo /etc/yum.repos.d/
-
-RUN yum install -y bazel4 --nogpgcheck
+RUN wget https://github.com/bazelbuild/bazelisk/releases/download/v1.11.0/bazelisk-linux-amd64 \
+    && mv bazelisk-linux-amd64 /usr/bin/bazel \
+    && chmod +x /usr/bin/bazel
 
 RUN mkdir /workspace
 

py/build_whl.sh

@@ -7,12 +7,6 @@ cd /workspace/Torch-TensorRT/py
 export CXX=g++
 export CUDA_HOME=/usr/local/cuda-11.3
 
-build_py36() {
-    /opt/python/cp36-cp36m/bin/python -m pip install -r requirements.txt
-    /opt/python/cp36-cp36m/bin/python setup.py bdist_wheel --release
-    #auditwheel repair --plat manylinux2014_x86_64
-}
-
 build_py37() {
     /opt/python/cp37-cp37m/bin/python -m pip install -r requirements.txt
     /opt/python/cp37-cp37m/bin/python setup.py bdist_wheel --release
@@ -31,7 +25,13 @@ build_py39() {
     #auditwheel repair --plat manylinux2014_x86_64
 }
 
-build_py36
+build_py10() {
+    /opt/python/cp310-cp310/bin/python -m pip install -r requirements.txt
+    /opt/python/cp310-cp310/bin/python setup.py bdist_wheel --release
+    #auditwheel repair --plat manylinux2014_x86_64
+}
+
 build_py37
 build_py38
 build_py39
+build_py310