Skip to content

Commit 274d8fa

Browse files
authored
Unbreak CI image build (tensorflow 2.6.5, ci_gpu bugfix) (#11546)
* Pin protobuf to 3.20.1 due to #11545. * Unpin and instead update to 2.6.5 * attempt to fix gpu build * Revert to 2.6.3, pin protobuf for ci-arm. * escape bash char
1 parent f31477f commit 274d8fa

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

docker/Dockerfile.ci_gpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ FROM nvidia/cuda:11.0.3-cudnn8-devel-ubuntu18.04
2424
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
2525

2626
# Base scripts
27-
RUN rm /etc/apt/sources.list.d/nvidia-ml.list && apt-get clean
27+
RUN rm -f /etc/apt/sources.list.d/nvidia-ml.list && apt-get clean
2828
RUN apt-get update --fix-missing
2929

3030
COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh

docker/install/ubuntu_install_tensorflow.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,4 +23,4 @@ set -o pipefail
2323
pip3 install \
2424
"h5py==3.1.0" \
2525
keras==2.6 \
26-
tensorflow==2.6.2
26+
tensorflow==2.6.5

docker/install/ubuntu_install_tensorflow_aarch64.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,5 +26,6 @@ apt-get install -y --no-install-recommends libhdf5-dev
2626
pip3 install \
2727
"h5py==3.1.0" \
2828
keras==2.6 \
29-
tensorflow-aarch64==2.6.2 \
29+
tensorflow-aarch64==2.6.3 \
30+
"protobuf<4" \
3031
-f https://snapshots.linaro.org/ldcg/python-cache/tensorflow-aarch64/

0 commit comments

Comments
 (0)