Skip to content
Permalink
Browse files

Upgrade NCCL to 2.5.6, unpin CUDA docker digest (#1600)

Signed-off-by: Travis Addair <taddair@uber.com>
  • Loading branch information
tgaddair committed Dec 19, 2019
1 parent 67b8363 commit fc49a721342d3c3e3acebbc21958206619c74aaa
Showing with 2 additions and 2 deletions.
  1. +2 −2 docker-compose.test.yml
@@ -273,9 +273,9 @@ services:
extends: test-gpu-base
build:
args:
CUDA_DOCKER_VERSION: 10.1-devel-ubuntu18.04@sha256:68efc9bbe07715c54ff30850aeb2e6f0d0b692af3c8dd40f13c0b179bfc0bc15
CUDA_DOCKER_VERSION: 10.1-devel-ubuntu18.04
CUDNN_VERSION: 7.6.5.32-1+cuda10.1
NCCL_VERSION_OVERRIDE: 2.4.8-1+cuda10.1
NCCL_VERSION_OVERRIDE: 2.5.6-1+cuda10.1
MPI_KIND: OpenMPI
PYTHON_VERSION: 3.6
TENSORFLOW_PACKAGE: tf-nightly-gpu==2.1.0.dev20191203

0 comments on commit fc49a72

Please sign in to comment.
You can’t perform that action at this time.