diff options
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.gpu')
-rw-r--r-- | tensorflow/tools/docker/Dockerfile.gpu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.gpu b/tensorflow/tools/docker/Dockerfile.gpu index 0571dd7391..e212d10290 100644 --- a/tensorflow/tools/docker/Dockerfile.gpu +++ b/tensorflow/tools/docker/Dockerfile.gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04 +FROM nvidia/cuda:8.0-cudnn6-runtime-ubuntu16.04 LABEL maintainer="Craig Citro <craigcitro@google.com>" |