You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by le...@apache.org on 2022/03/16 13:40:29 UTC
[tvm] 01/02: Upgrade tensorflow to version to 2.6.x
This is an automated email from the ASF dual-hosted git repository.
leandron pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit c573163cb0a7f67334f1c0cd95e80311c1852fe1
Author: Ophir Frish <op...@arm.com>
AuthorDate: Thu Jan 27 12:28:30 2022 +0200
Upgrade tensorflow to version to 2.6.x
Upgrade the following versions:
keras - from 2.4.3 to 2.6
tensorflow - from 2.4.2 to 2.6.2
h5py - from version < 3.0 to version 3.1.0
---
docker/install/ubuntu_install_tensorflow.sh | 6 +++---
docker/install/ubuntu_install_vitis_ai_packages_ci.sh | 2 +-
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/docker/install/ubuntu_install_tensorflow.sh b/docker/install/ubuntu_install_tensorflow.sh
index 8a51fbb..eaf89ff 100755
--- a/docker/install/ubuntu_install_tensorflow.sh
+++ b/docker/install/ubuntu_install_tensorflow.sh
@@ -21,6 +21,6 @@ set -u
set -o pipefail
pip3 install \
- "h5py<3.0" \
- keras==2.4.3 \
- tensorflow==2.4.2
+ "h5py==3.1.0" \
+ keras==2.6 \
+ tensorflow==2.6.2
diff --git a/docker/install/ubuntu_install_vitis_ai_packages_ci.sh b/docker/install/ubuntu_install_vitis_ai_packages_ci.sh
index 25c2140..269a957 100644
--- a/docker/install/ubuntu_install_vitis_ai_packages_ci.sh
+++ b/docker/install/ubuntu_install_vitis_ai_packages_ci.sh
@@ -23,7 +23,7 @@ set -o pipefail
export PYXIR_HOME=/opt/pyxir
mkdir "$PYXIR_HOME"
-pip3 install progressbar h5py==2.10.0
+pip3 install progressbar
git clone --recursive --branch v0.3.1 --depth 1 https://github.com/Xilinx/pyxir.git "${PYXIR_HOME}"
cd "${PYXIR_HOME}" && python3 setup.py install