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 77113c1d82..b2b39501cd 100644 --- a/tensorflow/tools/docker/Dockerfile.gpu +++ b/tensorflow/tools/docker/Dockerfile.gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:8.0-cudnn5-devel +FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04 MAINTAINER Craig Citro <craigcitro@google.com> |