From 56264ea4b72b31decf2b5e783630047eddc3771a Mon Sep 17 00:00:00 2001 From: Alexandre Lissy Date: Mon, 28 Sep 2020 10:23:00 +0200 Subject: [PATCH] Use correct 1.15.4 docker image --- Dockerfile.train.tmpl | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/Dockerfile.train.tmpl b/Dockerfile.train.tmpl index 05c09d4f..1342d054 100644 --- a/Dockerfile.train.tmpl +++ b/Dockerfile.train.tmpl @@ -1,6 +1,6 @@ # Please refer to the TRAINING documentation, "Basic Dockerfile for training" -FROM tensorflow/tensorflow:1.15.2-gpu-py3 +FROM tensorflow/tensorflow:1.15.4-gpu-py3 ENV DEBIAN_FRONTEND=noninteractive ENV DEEPSPEECH_REPO=#DEEPSPEECH_REPO# @@ -43,9 +43,6 @@ RUN pip3 install --upgrade native_client/ctcdecode/dist/*.whl # Prepare deps RUN pip3 install --upgrade pip==20.2.2 wheel==0.34.2 setuptools==49.6.0 -# FIXME: Hack because no 1.15.4 Docker image -RUN pip3 install --upgrade tensorflow-gpu==1.15.4 - # Install DeepSpeech # - No need for the decoder since we did it earlier # - There is already correct TensorFlow GPU installed on the base image,