diff --git a/README.md b/README.md index 6e8399c22a..78789a9451 100644 --- a/README.md +++ b/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 diff --git a/docker/Dockerfile b/docker/Dockerfile index bd43a17a02..d235408c4b 100644 --- a/docker/Dockerfile +++ b/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 \ diff --git a/docker/Dockerfile.docs b/docker/Dockerfile.docs index 1cc75ada4f..05429435ec 100644 --- a/docker/Dockerfile.docs +++ b/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 diff --git a/docker/Dockerfile.ngc b/docker/Dockerfile.ngc index 1e8e4b01c5..f145fa8ff2 100644 --- a/docker/Dockerfile.ngc +++ b/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 \ diff --git a/docker/setup_nox.sh b/docker/setup_nox.sh index 8b7e148f81..1f6d6e2006 100644 --- a/docker/setup_nox.sh +++ b/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