chore: Aligning bazel version for consistency across different docker…

README.md

@@ -111,7 +111,7 @@ torch.jit.save(trt_ts_module, "trt_torchscript_module.ts") # save the TRT embedd
 
 These are the following dependencies used to verify the testcases. Torch-TensorRT can work with other versions, but the tests are not guaranteed to pass.
 
-- Bazel 5.1.1
+- Bazel 5.2.0
 - Libtorch 1.12.0 (built with CUDA 11.3)
 - CUDA 11.3
 - cuDNN 8.4.1

docker/Dockerfile

@@ -9,7 +9,7 @@ RUN rm -rf /opt/pytorch/torch_tensorrt /usr/bin/bazel
 
 ARG ARCH="x86_64"
 ARG TARGETARCH="amd64"
-ARG BAZEL_VERSION=5.1.1
+ARG BAZEL_VERSION=5.2.0
 
 RUN [[ "$TARGETARCH" == "amd64" ]] && ARCH="x86_64" || ARCH="${TARGETARCH}" \
  && wget -q https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-linux-${ARCH} -O /usr/bin/bazel \

docker/Dockerfile.docs

@@ -1,10 +1,12 @@
 FROM nvcr.io/nvidia/tensorrt:22.06-py3
 
+ARG BAZEL_VERSION=5.2.0
+
 RUN curl https://bazel.build/bazel-release.pub.gpg | apt-key add -
 RUN echo "deb [arch=amd64] https://storage.googleapis.com/bazel-apt stable jdk1.8" | tee /etc/apt/sources.list.d/bazel.list
 
-RUN apt-get update && apt-get install -y bazel-5.2.0 clang-format-9 libjpeg9 libjpeg9-dev
-RUN ln -s /usr/bin/bazel-5.2.0 /usr/bin/bazel
+RUN apt-get update && apt-get install -y bazel-${BAZEL_VERSION} clang-format-9 libjpeg9 libjpeg9-dev
+RUN ln -s /usr/bin/bazel-${BAZEL_VERSION} /usr/bin/bazel
 
 # Workaround for bazel expecting both static and shared versions, we only use shared libraries inside container
 RUN cp /usr/lib/x86_64-linux-gnu/libnvinfer.so /usr/lib/x86_64-linux-gnu/libnvinfer_static.a

docker/Dockerfile.ngc

@@ -9,7 +9,7 @@ RUN rm -rf /opt/pytorch/torch_tensorrt /usr/bin/bazel
 
 ARG ARCH="x86_64"
 ARG TARGETARCH="amd64"
-ARG BAZEL_VERSION=5.1.1
+ARG BAZEL_VERSION=5.2.0
 
 RUN [[ "$TARGETARCH" == "amd64" ]] && ARCH="x86_64" || ARCH="${TARGETARCH}" \
  && wget -q https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-linux-${ARCH} -O /usr/bin/bazel \

docker/setup_nox.sh

@@ -7,7 +7,7 @@ set -e
 post=${1:-""}
 
 # fetch bazel executable
-BAZEL_VERSION=5.1.1
+BAZEL_VERSION=5.2.0
 ARCH=$(uname -m)
 if [[ "$ARCH" == "aarch64" ]]; then ARCH="arm64"; fi
 wget -q https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel-${BAZEL_VERSION}-linux-${ARCH} -O /usr/bin/bazel