diff --git a/Jenkins/Dockerfile.tf-torch-cpu b/Jenkins/Dockerfile.tf-torch-cpu index ca2bc34e4d8..95a6c9400ec 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 \ @@ -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 index ff7bd00924b..04eb6ba2067 100644 --- 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 index 6a240ec444a..261d096649a 100644 --- 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 index 7076aeaaaee..50da5040f0d 100644 --- 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 index 7e25ab984c7..0850b0fcc7e 100644 --- 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 index 162570aafe7..8c1a2a0f7fc 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 @@ -10,6 +10,7 @@ matplotlib>=3 networkx numpy<1.24,>=1.20.5 onnx==1.14.1 +onnxscript onnxsim osqp pandas==1.5.3 @@ -17,6 +18,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 566973a35ba..e2d07861d00 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 @@ -10,6 +10,7 @@ matplotlib>=3 networkx numpy<1.24,>=1.16.6 onnx==1.14.1 +onnxscript onnxsim osqp pandas==1.5.3 @@ -17,6 +18,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 f698120a8c5..f8543d31f5d 100644 --- 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