[circle-mlir/infra] Adjust the numpy version (#14746)

This PR adjusts the numpy version to ensure compatibility with Ubuntu 20.04 and 22.04. The numpy version `1.26.4` was not compatible with python-3.8, which is the default version on Ubuntu 20.04. ONE-DCO-1.0-Signed-off-by: Seungho Henry Park <[email protected]>
Samsung · Feb 25, 2025 · 50ba16a · 50ba16a
1 parent 89e8501
commit 50ba16a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/circle-mlir/infra/docker/u2204/Dockerfile b/circle-mlir/infra/docker/u2204/Dockerfile
@@ -27,7 +27,7 @@ RUN apt-get update \
 RUN apt-get update \
  && apt-get -qqy install python3 python3-pip python3-venv python3-dev python3-all dh-python
 RUN python3 -m pip install --upgrade pip setuptools
-RUN python3 -m pip install yapf==0.43.0 numpy==1.26.4 h5py==3.8.0 einops
+RUN python3 -m pip install yapf==0.43.0 numpy==1.24.4 h5py==3.8.0 einops
 
 # TODO upgrade
 ARG VER_TORCH=2.1.0

diff --git a/circle-mlir/infra/overlay/prepare-venv b/circle-mlir/infra/overlay/prepare-venv
@@ -62,4 +62,4 @@ ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install onnx==${VER_ONNX}
 ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install onnxruntime==${VER_ONNXRUNTIME}
 
 ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install h5py einops
-${VENV_PYTHON} -m pip ${PIP_OPTIONS} install numpy==1.26.4
+${VENV_PYTHON} -m pip ${PIP_OPTIONS} install numpy==1.24.4