aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/docker/Dockerfile.devel-gpu
diff options
context:
space:
mode:
authorGravatar Gunhan Gulsoy <gunan@google.com>2017-12-07 23:19:45 -0800
committerGravatar GitHub <noreply@github.com>2017-12-07 23:19:45 -0800
commitb43d0f3c98140edfebb8295ea4a4b661e2fc2a85 (patch)
tree1be014bf8d2650184f4aac63d35d4e9fe29a1e7c /tensorflow/tools/docker/Dockerfile.devel-gpu
parentf7d392da2a455ffcc5e410e9c7c0514a5062c218 (diff)
Upgrade cuda to 9 and cudnn version to 7. (#14773)
* Upgrade cuda to 9 and cudnn version to 7. * Also update XLA build script. * Fix typos. * Fix botched conflict resolution. * Disable transpose_op_test in opensource until the timeouts are fixed.
Diffstat (limited to 'tensorflow/tools/docker/Dockerfile.devel-gpu')
-rw-r--r--tensorflow/tools/docker/Dockerfile.devel-gpu2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu b/tensorflow/tools/docker/Dockerfile.devel-gpu
index 041f45971b..339a39bcf3 100644
--- a/tensorflow/tools/docker/Dockerfile.devel-gpu
+++ b/tensorflow/tools/docker/Dockerfile.devel-gpu
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04
+FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
LABEL maintainer="Craig Citro <craigcitro@google.com>"