Skip to content

Commit 06533fe

Browse files
committed
refactor!: Updating bazel version for py build container
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 <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 90946ae commit 06533fe

File tree

5 files changed

+7
-7
lines changed

5 files changed

+7
-7
lines changed

.bazelversion

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
4.0.0
1+
4.2.1

WORKSPACE

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ local_repository(
4141
new_local_repository(
4242
name = "cuda",
4343
build_file = "@//third_party/cuda:BUILD",
44-
path = "/usr/local/cuda-11.1/",
44+
path = "/usr/local/cuda-11.3/",
4545
)
4646

4747
new_local_repository(

docsrc/tutorials/installation.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ Torch-TensorRT is built with Bazel, so begin by installing it.
5252

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

238238
NVIDIA hosts builds the latest release branch for Jetson here:
239239

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

242242

243243
Enviorment Setup

py/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
1-
FROM pytorch/manylinux-cuda111
1+
FROM pytorch/manylinux-cuda113
22

33
RUN yum install -y ninja-build
44

5-
RUN wget https://copr.fedorainfracloud.org/coprs/vbatts/bazel/repo/epel-7/vbatts-bazel-epel-7.repo \
5+
RUN wget --no-check-certificate https://copr.fedorainfracloud.org/coprs/vbatts/bazel/repo/epel-7/vbatts-bazel-epel-7.repo \
66
&& mv vbatts-bazel-epel-7.repo /etc/yum.repos.d/
77

88
RUN yum install -y bazel4 --nogpgcheck

py/build_whl.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

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

5-
cd /workspace/TRTorch/py
5+
cd /workspace/Torch-TensorRT/py
66

77
export CXX=g++
88

0 commit comments

Comments
 (0)