fix pinning of numpy version in docker files by reinstalling specifie… (

#2085)
deepjavalibrary · Jun 18, 2024 · f48e1a7 · f48e1a7
1 parent b900358
commit f48e1a7
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/serving/docker/deepspeed.Dockerfile b/serving/docker/deepspeed.Dockerfile
@@ -17,6 +17,7 @@ ARG djl_version=0.26.0
 ARG python_version=3.10
 ARG torch_version=2.1.2
 ARG torch_vision_version=0.16.2
+ARG numpy_version=1.26.4
 # HF Deps
 ARG protobuf_version=3.20.3
 ARG transformers_version=4.36.2
@@ -108,6 +109,7 @@ RUN scripts/patch_oss_dlc.sh python && \
     useradd -m -d /home/djl djl && \
     chown -R djl:djl /opt/djl && \
     rm -rf scripts && \
+    pip3 install numpy==${numpy_version} && \
     pip3 cache purge && \
     apt-get clean -y && rm -rf /var/lib/apt/lists/*
 

diff --git a/serving/docker/pytorch-inf2.Dockerfile b/serving/docker/pytorch-inf2.Dockerfile
@@ -24,6 +24,7 @@ ARG accelerate_version=0.23.0
 ARG diffusers_version=0.22.0
 ARG pydantic_version=1.10.13
 ARG optimum_neuron_version=0.0.16
+ARG numpy_version=1.26.4
 EXPOSE 8080
 
 # Sets up Path for Neuron tools
@@ -73,6 +74,7 @@ RUN mkdir -p /opt/djl/bin && cp scripts/telemetry.sh /opt/djl/bin && \
     neuronx_distributed==${neuronx_distributed_version} protobuf==${protobuf_version} sentencepiece jinja2 \
     diffusers==${diffusers_version} opencv-contrib-python-headless  Pillow --extra-index-url=https://pip.repos.neuron.amazonaws.com \
     pydantic==${pydantic_version} optimum optimum-neuron==${optimum_neuron_version} && \
+    pip install numpy=${numpy_version} && \
     scripts/install_s5cmd.sh x64 && \
     scripts/patch_oss_dlc.sh python && \
     useradd -m -d /home/djl djl && \

diff --git a/serving/docker/tensorrt-llm.Dockerfile b/serving/docker/tensorrt-llm.Dockerfile
@@ -28,6 +28,7 @@ ARG triton_toolkit_wheel="https://publish.djl.ai/tritonserver/r23.11/tritontoolk
 ARG pydantic_version=1.10.13
 ARG ammo_version=0.5.0
 ARG pynvml_verison=11.4.1
+ARG numpy_version=1.26.4
 EXPOSE 8080
 
 COPY dockerd-entrypoint-with-cuda-compat.sh /usr/local/bin/dockerd-entrypoint.sh
@@ -106,6 +107,7 @@ RUN scripts/install_djl_serving.sh $djl_version && \
     useradd -m -d /home/djl djl && \
     chown -R djl:djl /opt/djl && \
     rm -rf scripts && \
+    pip3 install numpy==${numpy_version} && \
     pip3 cache purge && \
     apt-get clean -y && rm -rf /var/lib/apt/lists/*