fix: fixed dockerfile errors and bazel error

Signed-off-by: Anurag Dixit <[email protected]>

Author: andi4191

docker/Dockerfile

@@ -34,6 +34,7 @@ WORKDIR /workspace/torch_tensorrt/src
 RUN cp ./docker/WORKSPACE.docker WORKSPACE
 
 # This script builds both libtorchtrt bin/lib/include tarball and the Python wheel, in dist/
+RUN pip3 install --upgrade pip
 RUN ./docker/dist-build.sh
 
 FROM base as torch-tensorrt
@@ -45,7 +46,8 @@ RUN rm -rf /opt/pytorch/torch_tensorrt
 COPY . /workspace/torch_tensorrt
 COPY --from=torch-tensorrt-builder  /workspace/torch_tensorrt/src/py/dist/ .
 
-RUN pip3 install ipywidgets --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org
+RUN pip3 install --upgrade pip \
+  && pip3 install ipywidgets --trusted-host pypi.org --trusted-host pypi.python.org --trusted-host=files.pythonhosted.org
 RUN jupyter nbextension enable --py widgetsnbextension
 
 RUN pip3 install *.whl && rm -fr /workspace/torch_tensorrt/py/dist/* *.whl

py/setup.py

@@ -33,6 +33,7 @@
 __cudnn_version__ = "8.4"
 __tensorrt_version__ = "8.4"
 
+
 def get_git_revision_short_hash() -> str:
     return subprocess.check_output(["git", "rev-parse", "--short", "HEAD"]).decode("ascii").strip()
 

third_party/tensorrt/local/BUILD

@@ -351,11 +351,6 @@ cc_library(
         ":ci_rhel_x86_64_linux": glob(["lib64/libnvinfer_plugin.so"]),
         "//conditions:default": ["lib/x86_64-linux-gnu/libnvinfer_plugin.so"],
     }),
-    hdrs = select({
-        ":aarch64_linux": glob(["include/aarch64-linux-gnu/NvInferPlugin*.h"]),
-        ":windows": glob(["include/NvInferPlugin*.h"]),
-        "//conditions:default": glob(["include/x86_64-linux-gnu/NvInferPlugin*.h"]),
-    }),
     copts = [
         "-pthread",
     ],