Add missing dependencies (#3495)

Signed-off-by: Kyunggeun Lee <quic_kyunggeu@quicinc.com>
quic · Nov 16, 2024 · bcb8b8c · bcb8b8c
1 parent cdfd5eb
commit bcb8b8c
Show file tree

Hide file tree

Showing 8 changed files with 11 additions and 0 deletions.
diff --git a/Jenkins/Dockerfile.tf-torch-cpu b/Jenkins/Dockerfile.tf-torch-cpu
@@ -150,6 +150,7 @@ RUN python3 -m pip --no-cache-dir install \
         nbsphinx \
         'numpy<1.24,>=1.20.5' \
         onnx==1.12.0 \
+        onnxscript \
         onnxsim \
         onnxruntime==1.18.1 \
         onnxruntime-extensions \
@@ -169,6 +170,7 @@ RUN python3 -m pip --no-cache-dir install \
         pytest-cov \
         pytorch-ignite \
         PyYAML \
+        safetensors \
         scikit-learn==1.1.3 \
         scipy==1.8.1 \
         spconv==2.1.20 \

diff --git a/Jenkins/Dockerfile.torch-cpu b/Jenkins/Dockerfile.torch-cpu
@@ -181,6 +181,7 @@ RUN python3 -m pip --no-cache-dir install \
         pytest-cov \
         pytorch-ignite \
         PyYAML \
+        safetensors \
         scikit-learn==1.1.3 \
         scipy==1.8.1 \
         spconv \

diff --git a/Jenkins/Dockerfile.torch-cpu-pt113 b/Jenkins/Dockerfile.torch-cpu-pt113
@@ -181,6 +181,7 @@ RUN python3 -m pip --no-cache-dir install \
         pytest-cov \
         pytorch-ignite \
         PyYAML \
+        safetensors \
         scikit-learn==1.1.3 \
         scipy==1.8.1 \
         spconv==2.1.20 \

diff --git a/Jenkins/Dockerfile.torch-gpu b/Jenkins/Dockerfile.torch-gpu
@@ -193,6 +193,7 @@ RUN python3 -m pip --no-cache-dir install \
         pytest-cov \
         pytorch-ignite \
         PyYAML \
+        safetensors \
         scikit-learn==1.1.3 \
         scipy==1.8.1 \
         spconv-cu120 \

diff --git a/Jenkins/Dockerfile.torch-gpu-pt113 b/Jenkins/Dockerfile.torch-gpu-pt113
@@ -193,6 +193,7 @@ RUN python3 -m pip --no-cache-dir install \
         pytest-cov \
         pytorch-ignite \
         PyYAML \
+        safetensors \
         scikit-learn==1.1.3 \
         scipy==1.8.1 \
         spconv-cu117 \

diff --git a/packaging/dependencies/fast-release/torch-gpu-pt21-py310/reqs_pip_torch_gpu.txt b/packaging/dependencies/fast-release/torch-gpu-pt21-py310/reqs_pip_torch_gpu.txt
@@ -10,13 +10,15 @@ matplotlib>=3
 networkx
 numpy<1.24,>=1.20.5
 onnx==1.14.1
+onnxscript
 onnxsim
 osqp
 pandas==1.5.3
 protobuf==3.20.2
 psutil
 pybind11
 PyYAML
+safetensors
 scikit-learn==1.1.3
 scipy==1.8.1
 setuptools

diff --git a/packaging/dependencies/fast-release/torch-gpu-pt21-py38/reqs_pip_torch_gpu.txt b/packaging/dependencies/fast-release/torch-gpu-pt21-py38/reqs_pip_torch_gpu.txt
@@ -10,13 +10,15 @@ matplotlib>=3
 networkx
 numpy<1.24,>=1.16.6
 onnx==1.14.1
+onnxscript
 onnxsim
 osqp
 pandas==1.5.3
 protobuf==3.20.2
 psutil
 pybind11
 PyYAML
+safetensors
 scikit-learn==1.1.3
 scipy==1.8.1
 setuptools

diff --git a/packaging/dependencies/reqs_pip_torch_common.txt b/packaging/dependencies/reqs_pip_torch_common.txt
@@ -10,4 +10,5 @@ onnxsim
 pandas==1.5.3
 protobuf==3.20.2
 psutil
+safetensors
 tensorboardX==2.4