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
commit 2e5efe5e15
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -1,6 +1,6 @@
# 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
# We need to purge python3-xdg because it's breaking STT install later with