From 0b60e4dbbbc55af5a0a22badb50a551c306aa28a Mon Sep 17 00:00:00 2001
From: Reuben Morais <reuben.morais@gmail.com>
Date: Tue, 28 Sep 2021 13:38:11 +0200
Subject: [PATCH] Build KenLM in same base as final img

---
 Dockerfile.train | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.train b/Dockerfile.train
index 3fbf3dcb..dba45e6e 100644
--- a/Dockerfile.train
+++ b/Dockerfile.train
@@ -2,7 +2,7 @@
 # You can train "acoustic models" with audio + Tensorflow, and
 # you can create "scorers" with text + KenLM.
 
-FROM ubuntu:20.04 AS kenlm-build
+FROM nvcr.io/nvidia/tensorflow:20.06-tf1-py3 AS kenlm-build
 ENV DEBIAN_FRONTEND=noninteractive
 
 RUN apt-get update && \