Skip to content

Commit

Permalink
Merge pull request #1250 from pytorch/anuragd/bump_bazel_version
Browse files Browse the repository at this point in the history
chore: Aligning bazel version for consistency across different docker…
  • Loading branch information
andi4191 authored Aug 10, 2022
2 parents da7569e + 60da3da commit f814415
Show file tree
Hide file tree
Showing 5 changed files with 8 additions and 6 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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 \
Expand Down
6 changes: 4 additions & 2 deletions docker/Dockerfile.docs
Original file line number Diff line number Diff line change
@@ -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
Expand Down
2 changes: 1 addition & 1 deletion docker/Dockerfile.ngc
Original file line number Diff line number Diff line change
Expand Up @@ -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 \
Expand Down
2 changes: 1 addition & 1 deletion docker/setup_nox.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit f814415

Please sign in to comment.