From b354bdf692d11b4eeab75a87ae9af06d927effb5 Mon Sep 17 00:00:00 2001 From: Kyunggeun Lee Date: Fri, 15 Nov 2024 12:16:46 -0800 Subject: [PATCH 1/2] Add missing package dependencies Signed-off-by: Kyunggeun Lee --- .../fast-release/torch-gpu-pt21-py310/reqs_pip_torch_gpu.txt | 2 ++ .../fast-release/torch-gpu-pt21-py38/reqs_pip_torch_gpu.txt | 2 ++ packaging/dependencies/reqs_pip_torch_common.txt | 1 + 3 files changed, 5 insertions(+) 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 index 7799a7022f..8247f7f3e2 100644 --- 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 @@ -12,6 +12,7 @@ matplotlib>=3 networkx numpy<1.24,>=1.20.5 onnx==1.14.1 +onnxscript onnxsim osqp pandas==1.5.3 @@ -20,6 +21,7 @@ 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 index 35358ea1d7..ca28804407 100644 --- 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 @@ -11,6 +11,7 @@ matplotlib>=3 networkx numpy<1.24,>=1.16.6 onnx==1.14.1 +onnxscript onnxsim osqp pandas==1.5.3 @@ -19,6 +20,7 @@ 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 index f6298fe762..5b0e95dbe8 100644 --- a/packaging/dependencies/reqs_pip_torch_common.txt +++ b/packaging/dependencies/reqs_pip_torch_common.txt @@ -11,4 +11,5 @@ onnxsim pandas==1.5.3 protobuf==3.20.2 psutil +safetensors tensorboardX==2.4 From 8d418d16febd4a68e4913657ca088faf80971cbe Mon Sep 17 00:00:00 2001 From: Kyunggeun Lee Date: Fri, 15 Nov 2024 14:07:58 -0800 Subject: [PATCH 2/2] Add safetensors to Dockerfile Signed-off-by: Kyunggeun Lee --- Jenkins/Dockerfile.tf-torch-cpu | 2 ++ Jenkins/Dockerfile.torch-cpu | 1 + Jenkins/Dockerfile.torch-cpu-pt113 | 1 + Jenkins/Dockerfile.torch-gpu | 1 + Jenkins/Dockerfile.torch-gpu-pt113 | 1 + 5 files changed, 6 insertions(+) diff --git a/Jenkins/Dockerfile.tf-torch-cpu b/Jenkins/Dockerfile.tf-torch-cpu index 5bba4c5e9d..5decbcf5bf 100644 --- 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 \ @@ -170,6 +171,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 index 5dcf3d5c0f..eca6be2364 100644 --- a/Jenkins/Dockerfile.torch-cpu +++ b/Jenkins/Dockerfile.torch-cpu @@ -182,6 +182,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 index 17b2e578cf..7f8d730b43 100644 --- a/Jenkins/Dockerfile.torch-cpu-pt113 +++ b/Jenkins/Dockerfile.torch-cpu-pt113 @@ -182,6 +182,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 index 3790406d5b..db8df8da54 100644 --- a/Jenkins/Dockerfile.torch-gpu +++ b/Jenkins/Dockerfile.torch-gpu @@ -194,6 +194,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 index 2880493cd0..5571b3a346 100644 --- a/Jenkins/Dockerfile.torch-gpu-pt113 +++ b/Jenkins/Dockerfile.torch-gpu-pt113 @@ -194,6 +194,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 \