Skip to content

Commit

Permalink
[CI] Pin h5py version to < 3.0 to workaround issues with TF/Keras (ap…
Browse files Browse the repository at this point in the history
…ache#6808)

* Pin h5py to use the previous major release (2.x) and not
   new version 3.0, due to incompatibilities with TF and Keras
   that make TVMC and Frontend tests to fail
  • Loading branch information
leandron authored and Trevor Morris committed Dec 2, 2020
1 parent d5ed7e8 commit e62027d
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion docker/install/ubuntu_install_tensorflow.sh
Original file line number Diff line number Diff line change
Expand Up @@ -20,4 +20,7 @@ set -e
set -u
set -o pipefail

pip3 install tensorflow==2.3.1 keras==2.4.3 h5py
# h5py is pinned to minor than 3 due to issues with
# tensorflow:
# https://github.com/tensorflow/tensorflow/issues/44467
pip3 install tensorflow==2.3.1 keras==2.3.1 "h5py<3.0"

0 comments on commit e62027d

Please sign in to comment.