Skip to content

Commit 71e17da

Browse files
committed
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
1 parent d576f3d commit 71e17da

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

docker/install/ubuntu_install_tensorflow.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,6 @@ set -u
2121
set -o pipefail
2222

2323
pip3 install \
24-
"h5py<3.0" \
25-
keras==2.4.3 \
26-
tensorflow==2.4.2
24+
"h5py==3.1.0" \
25+
keras==2.6 \
26+
tensorflow==2.6.2

docker/install/ubuntu_install_vitis_ai_packages_ci.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ set -o pipefail
2323
export PYXIR_HOME=/opt/pyxir
2424
mkdir "$PYXIR_HOME"
2525

26-
pip3 install progressbar h5py==2.10.0
26+
pip3 install progressbar
2727

2828
git clone --recursive --branch v0.3.1 --depth 1 https://github.com/Xilinx/pyxir.git "${PYXIR_HOME}"
2929
cd "${PYXIR_HOME}" && python3 setup.py install

0 commit comments

Comments
 (0)