diff options
| author |  Toby Boyd <tobyboyd@google.com> | 2018-07-13 11:16:24 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-13 11:22:55 -0700 |
| commit | 3c3b9578bfcae1df34f7e885b086001fdb691993 (patch) | |
| tree | fcd9c612236fba269087f4cba0370aae76ec766f /tensorflow/tools/docker | |
| parent | 5e0dbc93eefca58644d4667ae6eeebb9da851d45 (diff) | |
Add NCCL 2.x to docker files.
PiperOrigin-RevId: 204500106
Diffstat (limited to 'tensorflow/tools/docker')
| -rw-r--r-- | tensorflow/tools/docker/Dockerfile.devel-gpu | 2 | ||||
| -rw-r--r-- | tensorflow/tools/docker/Dockerfile.gpu | 1 |
2 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/tools/docker/Dockerfile.devel-gpu b/tensorflow/tools/docker/Dockerfile.devel-gpu index 5ec43b8cb8..2818b822b8 100644 --- a/tensorflow/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow/tools/docker/Dockerfile.devel-gpu @@ -15,6 +15,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ git \ libcudnn7=7.1.4.18-1+cuda9.0 \ libcudnn7-dev=7.1.4.18-1+cuda9.0 \ + libnccl2=2.2.13-1+cuda9.0 \ + libnccl-dev=2.2.13-1+cuda9.0 \ libcurl3-dev \ libfreetype6-dev \ libhdf5-serial-dev \ diff --git a/tensorflow/tools/docker/Dockerfile.gpu b/tensorflow/tools/docker/Dockerfile.gpu index 9197651ff4..28d4371da3 100644 --- a/tensorflow/tools/docker/Dockerfile.gpu +++ b/tensorflow/tools/docker/Dockerfile.gpu @@ -13,6 +13,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ cuda-cusparse-9-0 \ curl \ libcudnn7=7.1.4.18-1+cuda9.0 \ + libnccl2=2.2.13-1+cuda9.0 \ libfreetype6-dev \ libhdf5-serial-dev \ libpng12-dev \ |