From 837982fa492cb4c9fe9514d2ad1be1aab0cf37de Mon Sep 17 00:00:00 2001 From: Ayush Dubey Date: Thu, 3 Oct 2019 15:27:36 -0700 Subject: [PATCH] fix --- perfzero/docker/Dockerfile_ubuntu_1804_tf_v2 | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/perfzero/docker/Dockerfile_ubuntu_1804_tf_v2 b/perfzero/docker/Dockerfile_ubuntu_1804_tf_v2 index 2fdb9f1c..7bd56a37 100644 --- a/perfzero/docker/Dockerfile_ubuntu_1804_tf_v2 +++ b/perfzero/docker/Dockerfile_ubuntu_1804_tf_v2 @@ -1,5 +1,5 @@ # Ubuntu 18.04 Python3 with CUDA 10 and the following: -# - Installs tf-nightly (this is TF 2.0) +# - Installs tf-nightly-gpu (this is TF 2.0) # - Installs requirements.txt for tensorflow/models FROM nvidia/cuda:10.0-base-ubuntu18.04 as base