diff options
| author |  Amit Patankar <amitpatankar@google.com> | 2018-07-19 17:28:52 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-19 17:32:58 -0700 |
| commit | 95ec73a1c8b5174ec2221c2b5ecaf179c9deef48 (patch) | |
| tree | 361db332cb93b2df2aae5357c15df46a123e6d30 /tensorflow | |
| parent | 78121905a545ca8e91ab1dbc899b5509933331d7 (diff) | |
Fix clang builds for NCCL error.
PiperOrigin-RevId: 205330050
Diffstat (limited to 'tensorflow')
| -rw-r--r-- | tensorflow/tools/ci_build/Dockerfile.gpu | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/tools/ci_build/Dockerfile.gpu b/tensorflow/tools/ci_build/Dockerfile.gpu index 7591ecc04e..6ab703d93d 100644 --- a/tensorflow/tools/ci_build/Dockerfile.gpu +++ b/tensorflow/tools/ci_build/Dockerfile.gpu @@ -14,6 +14,11 @@ RUN /install/install_bootstrap_deb_packages.sh RUN add-apt-repository -y ppa:openjdk-r/ppa && \ add-apt-repository -y ppa:george-edison55/cmake-3.x RUN /install/install_deb_packages.sh + +# Install NCCL +RUN apt-get update && apt-get install -y --no-install-recommends \ + libnccl2=2.2.13-1+cuda9.0 + RUN /install/install_pip_packages.sh RUN /install/install_bazel.sh RUN /install/install_golang.sh |