Merge pull request #1980 from coqui-ai/docker-kenlm-base

Build KenLM in same base as final image
coqui-ai · Sep 28, 2021 · 178cdac · 178cdac
2 parents 5691d4e + 0b60e4d
commit 178cdac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 && \