diff options
| author | Felix Abecassis <fabecassis@nvidia.com> | 2016-05-16 14:32:34 -0700 |
|---|---|---|
| committer | Felix Abecassis <fabecassis@nvidia.com> | 2016-05-16 14:32:34 -0700 |
| commit | 8730b146b7e19af189b9086e59fd1d5bc4214698 (patch) | |
| tree | 5bfa17ee2ff97ee50b33f555a07e436dcb41a63d /docker | |
| parent | b43c8e43a95608a00033f8f8867d32a201e5eed4 (diff) | |
| download | caffeonacl-8730b146b7e19af189b9086e59fd1d5bc4214698.tar.gz caffeonacl-8730b146b7e19af189b9086e59fd1d5bc4214698.tar.bz2 caffeonacl-8730b146b7e19af189b9086e59fd1d5bc4214698.zip | |
Update Dockerfile to cuDNN v5
Diffstat (limited to 'docker')
| -rw-r--r-- | docker/Makefile | 2 | ||||
| -rw-r--r-- | docker/standalone/gpu/Dockerfile | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/docker/Makefile b/docker/Makefile index 0de887d0..3a6575b0 100644 --- a/docker/Makefile +++ b/docker/Makefile @@ -22,7 +22,7 @@ docker_files: standalone_files standalone_files: standalone/cpu/Dockerfile standalone/gpu/Dockerfile -FROM_GPU = "nvidia/cuda:7.5-cudnn4-devel-ubuntu14.04" +FROM_GPU = "nvidia/cuda:7.5-cudnn5-devel-ubuntu14.04" FROM_CPU = "ubuntu:14.04" GPU_CMAKE_ARGS = -DUSE_CUDNN=1 CPU_CMAKE_ARGS = -DCPU_ONLY=1 diff --git a/docker/standalone/gpu/Dockerfile b/docker/standalone/gpu/Dockerfile index 371aad5b..daf6a722 100644 --- a/docker/standalone/gpu/Dockerfile +++ b/docker/standalone/gpu/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:7.5-cudnn4-devel-ubuntu14.04 +FROM nvidia/cuda:7.5-cudnn5-devel-ubuntu14.04 MAINTAINER caffe-maint@googlegroups.com RUN apt-get update && apt-get install -y --no-install-recommends \ |