[ci] fix pinning numpy inf2 error (#2101)

deepjavalibrary · Jun 25, 2024 · 972f509 · 972f509
1 parent f48e1a7
commit 972f509
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/serving/docker/pytorch-inf2.Dockerfile b/serving/docker/pytorch-inf2.Dockerfile
@@ -74,7 +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} && \
+    pip install numpy==${numpy_version} && \
     scripts/install_s5cmd.sh x64 && \
     scripts/patch_oss_dlc.sh python && \
     useradd -m -d /home/djl djl && \