Skip to content

Commit d59a766

Browse files
committed
remove package vulns
Signed-off-by: Srikanth Ramakrishna <srikanth.ramakrishna@intel.com>
1 parent e4a0073 commit d59a766

File tree

3 files changed

+7
-1
lines changed

3 files changed

+7
-1
lines changed

pytorch/Dockerfile

+2
Original file line numberDiff line numberDiff line change
@@ -320,6 +320,8 @@ ARG GID=109
320320
RUN groupadd -g ${GID} render &&\
321321
usermod -aG video,render model-server
322322

323+
RUN python -m pip install pip==23.3 setuptools==70.0.0
324+
323325
USER model-server
324326

325327
WORKDIR /home/model-server

pytorch/docker-compose.yaml

+2
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,8 @@ services:
183183
dependency.apt.openjdk-17-jdk: true
184184
dependency.apt.xpu-smi: ${XPU_SMI_VERSION:-1.2.38}
185185
dependency.idp: false
186+
dependency.pip.pip: 23.3
187+
dependency.pip.setuptools: 70.0.0
186188
dependency.python.pip: serving/torchserve-xpu-requirements.txt
187189
docs: serving
188190
org.opencontainers.base.name: "intel/python:3.10-core"

pytorch/serving/torchserve-xpu-requirements.txt

+3-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ torchvision==0.18.1+cxx11.abi
33
torchaudio==2.3.1+cxx11.abi
44
intel_extension_for_pytorch==2.3.110+xpu
55
--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us
6-
numpy==1.26.4
6+
numpy==2.1.1
77
captum>=0.7.0
88
cython>=3.0.10
99
pynvml>=11.5.0
@@ -12,3 +12,5 @@ pyyaml>=6.0.1
1212
torch-model-archiver==0.11.1
1313
torch-workflow-archiver==0.2.14
1414
torchserve==0.11.1
15+
pip==23.3
16+
setuptools==70.0.0

0 commit comments

Comments
 (0)