Skip to content

Commit 7620acc

Browse files
committed
chore: updates
1 parent 89f04db commit 7620acc

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

packaging/pre_build_script.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ export LD_LIBRARY_PATH=/opt/torch-tensorrt-builds/TensorRT-10.0.0.6/lib:$LD_LIBR
1010
wget https://github.com/bazelbuild/bazelisk/releases/download/v1.17.0/bazelisk-linux-amd64 \
1111
&& mv bazelisk-linux-amd64 /usr/bin/bazel \
1212
&& chmod +x /usr/bin/bazel
13-
13+
python -m pip install -r py/requirements.txt
1414
export TORCH_BUILD_NUMBER=$(python -c "import torch, urllib.parse as ul; print(ul.quote_plus(torch.__version__))")
1515

1616
cat toolchains/ci_workspaces/WORKSPACE.x86_64.release.rhel.tmpl | envsubst > WORKSPACE

py/requirements.txt

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,5 +5,10 @@ pybind11==2.6.2
55
torch==2.3.0
66
torchvision==0.18.0
77
--extra-index-url https://pypi.ngc.nvidia.com
8-
tensorrt>=10.0
98
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

0 commit comments

Comments
 (0)