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

Merge pull request #1889 from JRMeyer/dockerfile 

Use NVIDIA image in Dockerfile
This commit is contained in:
Reuben Morais 2021-06-10 16:01:52 +02:00 committed by GitHub
parent bf07f35420 38e06e4635
commit 2e5efe5e15
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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