Skip to content

Commit

Permalink
refactor!: Updating bazel version for py build container
Browse files Browse the repository at this point in the history
BREAKING CHANGE: This change updates the bazel version
to build Torch-TensorRT to 4.2.1.

This was done since the only version of bazel available
in our build container for python apis is 4.2.1

Signed-off-by: Naren Dasan <naren@narendasan.com>
Signed-off-by: Naren Dasan <narens@nvidia.com>
  • Loading branch information
narendasan committed Nov 5, 2021
1 parent 90946ae commit 06533fe
Show file tree
Hide file tree
Showing 5 changed files with 7 additions and 7 deletions.
2 changes: 1 addition & 1 deletion .bazelversion
@@ -1 +1 @@
4.0.0
4.2.1
2 changes: 1 addition & 1 deletion WORKSPACE
Expand Up @@ -41,7 +41,7 @@ local_repository(
new_local_repository(
name = "cuda",
build_file = "@//third_party/cuda:BUILD",
path = "/usr/local/cuda-11.1/",
path = "/usr/local/cuda-11.3/",
)

new_local_repository(
Expand Down
4 changes: 2 additions & 2 deletions docsrc/tutorials/installation.rst
Expand Up @@ -52,7 +52,7 @@ Torch-TensorRT is built with Bazel, so begin by installing it.

.. code-block:: shell
export BAZEL_VERSION=$(cat <PATH_TO_TRTORCH_ROOT>/.bazelversion)
export BAZEL_VERSION=$(cat <PATH_TO_TORCHTRT_ROOT>/.bazelversion)
mkdir bazel
cd bazel
curl -fSsL -O https://github.com/bazelbuild/bazel/releases/download/$BAZEL_VERSION/bazel-$BAZEL_VERSION-dist.zip
Expand Down Expand Up @@ -237,7 +237,7 @@ Install or compile a build of PyTorch/LibTorch for aarch64

NVIDIA hosts builds the latest release branch for Jetson here:

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-9-0-now-available/72048
https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048


Enviorment Setup
Expand Down
4 changes: 2 additions & 2 deletions py/Dockerfile
@@ -1,8 +1,8 @@
FROM pytorch/manylinux-cuda111
FROM pytorch/manylinux-cuda113

RUN yum install -y ninja-build

RUN wget https://copr.fedorainfracloud.org/coprs/vbatts/bazel/repo/epel-7/vbatts-bazel-epel-7.repo \
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
Expand Down
2 changes: 1 addition & 1 deletion py/build_whl.sh
Expand Up @@ -2,7 +2,7 @@

# Example usage: docker run -it -v$(pwd)/..:/workspace/TRTorch build_trtorch_wheel /bin/bash /workspace/TRTorch/py/build_whl.sh

cd /workspace/TRTorch/py
cd /workspace/Torch-TensorRT/py

export CXX=g++

Expand Down

0 comments on commit 06533fe

Please sign in to comment.