From 32b44c5447abd7d974549e46710e850beb5a7595 Mon Sep 17 00:00:00 2001 From: Reuben Morais Date: Thu, 19 Aug 2021 11:27:15 +0200 Subject: [PATCH] Downgrade training Dockerfile base image to one that has TFLite support See https://github.com/NVIDIA/tensorflow/issues/16 --- Dockerfile.train | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile.train b/Dockerfile.train index a31f17b3..0e86fa2a 100644 --- a/Dockerfile.train +++ b/Dockerfile.train @@ -44,7 +44,7 @@ RUN wget --no-check-certificate https://github.com/coqui-ai/STT/releases/downloa rm temp.tar.xz -FROM nvcr.io/nvidia/tensorflow:21.05-tf1-py3 +FROM nvcr.io/nvidia/tensorflow:20.06-tf1-py3 ENV DEBIAN_FRONTEND=noninteractive # We need to purge python3-xdg because