experiments
/
STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use NVIDIA image in Dockerfile

This commit is contained in:
Josh Meyer 2021-06-10 09:47:15 -04:00
parent bf07f35420
commit 38e06e4635
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile.train
View File

@ -1,6 +1,6 @@
# Please refer to the TRAINING documentation, "Basic Dockerfile for training" # Please refer to the TRAINING documentation, "Basic Dockerfile for training"
FROM tensorflow/tensorflow:1.15.4-gpu-py3 FROM nvcr.io/nvidia/tensorflow:21.05-tf1-py3
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
# We need to purge python3-xdg because it's breaking STT install later with # We need to purge python3-xdg because it's breaking STT install later with