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 <[email protected]> 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
