This is an automated email from the ASF dual-hosted git repository.
yongwww 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 3e386fd373 [CI] Upgrade `oneflow==0.9.0` (#17503)
3e386fd373 is described below
commit 3e386fd3732778d5eb7998f23e2d940741102d9a
Author: Masahiro Hiramori <[email protected]>
AuthorDate: Tue Nov 5 10:45:14 2024 +0900
[CI] Upgrade `oneflow==0.9.0` (#17503)
* Upgrade oneflow==0.9.0
* fix duplicate cmake installation
* fix pip config
---
docker/Dockerfile.ci_gpu | 3 +--
docker/install/ubuntu_install_oneflow.sh | 6 +++---
2 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu
index 7f5a68911c..cc9e488317 100644
--- a/docker/Dockerfile.ci_gpu
+++ b/docker/Dockerfile.ci_gpu
@@ -48,9 +48,8 @@ RUN bash /install/ubuntu_install_python.sh 3.9
ENV PATH ${TVM_VENV}/bin:$PATH
ENV PYTHONNOUSERSITE 1 # Disable .local directory from affecting CI.
-COPY install/ubuntu_install_cmake_source.sh
/install/ubuntu_install_cmake_source.sh
# Globally disable pip cache
-RUN bash /install/ubuntu_install_cmake_source.sh
+RUN pip config set global.no-cache-dir false
COPY install/ubuntu2204_install_llvm.sh /install/ubuntu2204_install_llvm.sh
RUN bash /install/ubuntu2204_install_llvm.sh
diff --git a/docker/install/ubuntu_install_oneflow.sh
b/docker/install/ubuntu_install_oneflow.sh
index 6c0e90e4f2..da2943e2f9 100755
--- a/docker/install/ubuntu_install_oneflow.sh
+++ b/docker/install/ubuntu_install_oneflow.sh
@@ -20,6 +20,6 @@ set -e
set -u
set -o pipefail
-pip3 install flowvision==0.1.0
-
-python3 -m pip install oneflow==0.8.0
+pip3 install \
+ oneflow==0.9.0 \
+ flowvision==0.1.0
