File tree Expand file tree Collapse file tree 4 files changed +12
-5
lines changed
Expand file tree Collapse file tree 4 files changed +12
-5
lines changed Original file line number Diff line number Diff line change @@ -32,6 +32,7 @@ ARG VER_TORCH=2.7.0+cpu
3232ARG VER_ONNX=1.18.0
3333ARG VER_ONNXRUNTIME=1.21.1
3434ARG VER_NUMPY=1.26.4
35+ ARG VER_H5PY=3.12.1
3536
3637# Clean archives (to reduce image size)
3738RUN apt-get clean -y
@@ -60,7 +61,7 @@ RUN ${WORK_DIR}/venv/bin/python3 -m pip install --upgrade pip setuptools
6061RUN ${WORK_DIR}/venv/bin/python3 -m pip install numpy==${VER_NUMPY} torch==${VER_TORCH} \
6162 -f https://download.pytorch.org/whl/torch
6263RUN ${WORK_DIR}/venv/bin/python3 -m pip install onnx==${VER_ONNX} onnxruntime==${VER_ONNXRUNTIME} \
63- h5py==3.8.0 einops
64+ h5py==${VER_H5PY} einops
6465
6566# Download experimental patch file for onnx-mlir
6667
Original file line number Diff line number Diff line change @@ -29,6 +29,7 @@ ARG VER_TORCH=2.7.0+cpu
2929ARG VER_ONNX=1.18.0
3030ARG VER_ONNXRUNTIME=1.21.1
3131ARG VER_NUMPY=1.26.4
32+ ARG VER_H5PY=3.12.1
3233
3334# Clean archives (to reduce image size)
3435RUN apt-get clean -y
@@ -52,7 +53,7 @@ RUN ${WORK_DIR}/venv/bin/python3 -m pip install --upgrade pip setuptools
5253RUN ${WORK_DIR}/venv/bin/python3 -m pip install numpy==${VER_NUMPY} torch==${VER_TORCH} \
5354 -f https://download.pytorch.org/whl/torch
5455RUN ${WORK_DIR}/venv/bin/python3 -m pip install onnx==${VER_ONNX} onnxruntime==${VER_ONNXRUNTIME} \
55- h5py==3.8.0 einops
56+ h5py==${VER_H5PY} einops
5657
5758# Download experimental patch file for onnx-mlir
5859
Original file line number Diff line number Diff line change @@ -29,6 +29,7 @@ ARG VER_TORCH=2.7.0+cpu
2929ARG VER_ONNX=1.18.0
3030ARG VER_ONNXRUNTIME=1.21.1
3131ARG VER_NUMPY=1.26.4
32+ ARG VER_H5PY=3.12.1
3233
3334# Clean archives (to reduce image size)
3435RUN apt-get clean -y
@@ -52,7 +53,7 @@ RUN ${WORK_DIR}/venv/bin/python3 -m pip install --upgrade pip setuptools
5253RUN ${WORK_DIR}/venv/bin/python3 -m pip install numpy==${VER_NUMPY} torch==${VER_TORCH} \
5354 -f https://download.pytorch.org/whl/torch
5455RUN ${WORK_DIR}/venv/bin/python3 -m pip install onnx==${VER_ONNX} onnxruntime==${VER_ONNXRUNTIME} \
55- h5py==3.8.0 einops
56+ h5py==${VER_H5PY} einops
5657
5758# Download experimental patch file for onnx-mlir
5859
Original file line number Diff line number Diff line change 2020# use +e as python3.10 may not exist in the system and 'command' will return error.
2121set +e
2222
23- PYTHON_CANDIDATES=(" python3.12" " python3.10" " python3" )
23+ PYTHON_CANDIDATES=(" python3.13 " " python3. 12" " python3.10" " python3" )
2424for py in " ${PYTHON_CANDIDATES[@]} " ; do
2525 PYTHON3_EXEC=$( command -v " $py " )
2626 if [[ -n " ${PYTHON3_EXEC} " ]]; then
@@ -56,6 +56,7 @@ VER_TORCH=2.7.0+cpu
5656VER_ONNX=1.18.0
5757VER_ONNXRUNTIME=1.21.1
5858VER_NUMPY=1.26.4
59+ VER_H5PY=3.12.1
5960
6061PYTHON_VERSION=$( $VENV_PYTHON -c " import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')" )
6162echo " python3 version is $PYTHON_VERSION "
@@ -66,6 +67,9 @@ if [[ "$PYTHON_VERSION" == "3.10" ]]; then
6667elif [[ " $PYTHON_VERSION " == " 3.12" ]]; then
6768 # Ubuntu 24.04 is python3.12
6869 : # use as is
70+ elif [[ " $PYTHON_VERSION " == " 3.13" ]]; then
71+ # Ubuntu 25.04 is python3.13
72+ : # use as is
6973else
7074 echo " Error prepare-venv: Unsupported python version: $PYTHON_VERSION "
7175 exit 1
9094PYTHON_PACKAGES=" torch==${VER_TORCH} "
9195PYTHON_PACKAGES+=" onnx==${VER_ONNX} "
9296PYTHON_PACKAGES+=" onnxruntime==${VER_ONNXRUNTIME} "
93- PYTHON_PACKAGES+=" h5py==3.8.0 "
97+ PYTHON_PACKAGES+=" h5py==${VER_H5PY} "
9498PYTHON_PACKAGES+=" einops "
9599PYTHON_PACKAGES+=" numpy==${VER_NUMPY} "
96100
You can’t perform that action at this time.
0 commit comments