File tree Expand file tree Collapse file tree 2 files changed +7
-2
lines changed Expand file tree Collapse file tree 2 files changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -10,7 +10,7 @@ export LD_LIBRARY_PATH=/opt/torch-tensorrt-builds/TensorRT-10.0.0.6/lib:$LD_LIBR
10
10
wget https://github.com/bazelbuild/bazelisk/releases/download/v1.17.0/bazelisk-linux-amd64 \
11
11
&& mv bazelisk-linux-amd64 /usr/bin/bazel \
12
12
&& chmod +x /usr/bin/bazel
13
-
13
+ python -m pip install -r py/requirements.txt
14
14
export TORCH_BUILD_NUMBER=$( python -c " import torch, urllib.parse as ul; print(ul.quote_plus(torch.__version__))" )
15
15
16
16
cat toolchains/ci_workspaces/WORKSPACE.x86_64.release.rhel.tmpl | envsubst > WORKSPACE
Original file line number Diff line number Diff line change @@ -5,5 +5,10 @@ pybind11==2.6.2
5
5
torch == 2.3.0
6
6
torchvision == 0.18.0
7
7
--extra-index-url https://pypi.ngc.nvidia.com
8
- tensorrt >= 10.0
9
8
pyyaml
9
+ # Extras
10
+ ninja >= 1.11.0
11
+ cffi >= 1.15.1
12
+ pybind11 == 2.6.2
13
+ wheel >= 0.40.0
14
+ future >= 0.18.3
You can’t perform that action at this time.
0 commit comments