From 972f509cf4945bae1918fe01077ca0bb19a8d617 Mon Sep 17 00:00:00 2001 From: Tyler Osterberg Date: Tue, 25 Jun 2024 14:54:19 -0700 Subject: [PATCH] [ci] fix pinning numpy inf2 error (#2101) --- serving/docker/pytorch-inf2.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/serving/docker/pytorch-inf2.Dockerfile b/serving/docker/pytorch-inf2.Dockerfile index 32ff30fce..cfcbfa4b7 100644 --- 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 && \