Merge pull request #1889 from JRMeyer/dockerfile
Use NVIDIA image in Dockerfile
This commit is contained in:
commit
2e5efe5e15
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue