This is an automated email from the ASF dual-hosted git repository.
tlopex pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 2a448ce4bb [CI] Update Pillow, pytest-rerunfailures, junitparser,
xgboost, onnx and pytorch (#18777)
2a448ce4bb is described below
commit 2a448ce4bb715074fb876bbb7bc1133a99d80d7e
Author: Masahiro Hiramori <[email protected]>
AuthorDate: Sat Feb 14 23:16:12 2026 +0900
[CI] Update Pillow, pytest-rerunfailures, junitparser, xgboost, onnx and
pytorch (#18777)
Fix https://github.com/apache/tvm/pull/18775#issuecomment-3901181065 ,
update xgboost, onnx and pytorch
---
docker/install/ubuntu_install_onnx.sh | 65 ++++---------------------
docker/install/ubuntu_install_python_package.sh | 6 +--
docker/install/ubuntu_install_redis.sh | 2 +-
3 files changed, 14 insertions(+), 59 deletions(-)
diff --git a/docker/install/ubuntu_install_onnx.sh
b/docker/install/ubuntu_install_onnx.sh
index dcf2f2c421..72cf952107 100755
--- a/docker/install/ubuntu_install_onnx.sh
+++ b/docker/install/ubuntu_install_onnx.sh
@@ -20,13 +20,6 @@ set -e
set -u
set -o pipefail
-# We need to fix the onnx version because changing versions tends to break
tests
-# TODO(mbrookhart): periodically update
-
-# onnx 1.9 removed onnx optimizer from the main repo (see
-# https://github.com/onnx/onnx/pull/2834). When updating the CI image
-# to onnx>=1.9, onnxoptimizer should also be installed.
-
# Get the Python version
PYTHON_VERSION=$(python3 -c "import sys;
print(f'{sys.version_info.major}.{sys.version_info.minor}')")
@@ -34,56 +27,18 @@ PYTHON_VERSION=$(python3 -c "import sys;
print(f'{sys.version_info.major}.{sys.v
DEVICE=${1:-cpu}
# Install the onnx package
-pip3 install future
-
-if [ "$PYTHON_VERSION" == "3.9" ]; then
- pip3 install \
- onnx==1.16.0 \
- onnxruntime==1.19.2 \
- onnxoptimizer==0.2.7
+pip3 install \
+ onnx==1.20.1 \
+ onnxruntime==1.23.2 \
+ onnxoptimizer==0.4.2
- if [ "$DEVICE" == "cuda" ]; then
- pip3 install \
- torch==2.7.0 \
- torchvision==0.22.0 \
- --index-url https://download.pytorch.org/whl/cu118
- else
- pip3 install \
- torch==2.7.0 \
- torchvision==0.22.0 \
- --extra-index-url https://download.pytorch.org/whl/cpu
- fi
-elif [ "$PYTHON_VERSION" == "3.11" ]; then
+if [ "$DEVICE" == "cuda" ]; then
pip3 install \
- onnx==1.17.0 \
- onnxruntime==1.20.1 \
- onnxoptimizer==0.2.7
-
- if [ "$DEVICE" == "cuda" ]; then
- pip3 install \
- torch==2.7.0 \
- torchvision==0.22.0 \
- --index-url https://download.pytorch.org/whl/cu118
- else
- pip3 install \
- torch==2.7.0 \
- torchvision==0.22.0 \
- --extra-index-url https://download.pytorch.org/whl/cpu
- fi
+ torch==2.10.0 \
+ torchvision==0.25.0
else
pip3 install \
- onnx==1.12.0 \
- onnxruntime==1.12.1 \
- onnxoptimizer==0.2.7
-
- if [ "$DEVICE" == "cuda" ]; then
- pip3 install \
- torch==2.4.1 \
- torchvision==0.19.1
- else
- pip3 install \
- torch==2.4.1 \
- torchvision==0.19.1 \
- --extra-index-url https://download.pytorch.org/whl/cpu
- fi
+ torch==2.10.0 \
+ torchvision==0.25.0 \
+ --extra-index-url https://download.pytorch.org/whl/cpu
fi
diff --git a/docker/install/ubuntu_install_python_package.sh
b/docker/install/ubuntu_install_python_package.sh
index c0aa1f6b54..819e1dac02 100755
--- a/docker/install/ubuntu_install_python_package.sh
+++ b/docker/install/ubuntu_install_python_package.sh
@@ -29,17 +29,17 @@ pip3 install --upgrade \
numpy==1.21.* \
orderedset \
packaging \
- Pillow==9.1.0 \
+ Pillow==12.1.1 \
psutil \
pytest \
git+https://github.com/tlc-pack/tlcpack-sphinx-addon.git@768ec1dce349fe4708f6ad68be1ebb3f3dabafa1
\
pytest-profiling \
pytest-xdist \
- pytest-rerunfailures==10.2 \
+ pytest-rerunfailures==16.1 \
requests \
scipy \
Jinja2 \
- junitparser==2.4.2 \
+ junitparser==4.0.2 \
six \
tornado \
"ml_dtypes>=0.5.1"
diff --git a/docker/install/ubuntu_install_redis.sh
b/docker/install/ubuntu_install_redis.sh
index 72049f7b9a..240df8715d 100755
--- a/docker/install/ubuntu_install_redis.sh
+++ b/docker/install/ubuntu_install_redis.sh
@@ -22,4 +22,4 @@ set -o pipefail
apt-get update && apt-install-and-clear -y redis-server
pip3 install \
- xgboost==1.4.2
+ xgboost==3.2.0
