Skip to content

Commit 49d4c2d

Browse files
author
Thiago Crepaldi
committed
Update OOBE GPU Dockerfile to CUDA 10
1 parent f48e11c commit 49d4c2d

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

Tests/Install/linux/Dockerfile-Ubuntu16-GPU

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,14 @@
1-
FROM nvidia/cuda:8.0-runtime-ubuntu16.04
2-
# use nvidia-docker to run!
1+
# Tag: nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
2+
# Created: 2018-10-22T21:14:30.605789926Z
3+
# Label: com.nvidia.cuda.version: 10.0.
4+
# Label: com.nvidia.cudnn.version: 7.3.1.20
5+
# Label: com.nvidia.nccl.version: 2.3.5
6+
#
7+
# To build, run from the parent with the command line:
8+
# docker build -t <image name> -f CNTK-GPU-Image/Dockerfile .
9+
10+
# Ubuntu 16.04.5
11+
FROM nvidia/cuda@sha256:362e4e25aa46a18dfa834360140e91b61cdb0a3a2796c8e09dadb268b9de3f6b
312

413
ARG PY_VERSION
514
ARG WHEEL_BASE_URL

0 commit comments

Comments
 (0)