Merge pull request #13262 from typhoonzero/fix_cuda9_docker_build

fix cuda9 docker build
7 years ago · b6bddd2c29
parent b805751598 04106cb1aa
commit b6bddd2c29
1 changed files with 1 additions and 1 deletions
--- a/paddle/scripts/paddle_build.sh
+++ b/paddle/scripts/paddle_build.sh
@ -500,7 +500,7 @@ EOF
 EOF

    if [[ ${WITH_GPU} == "ON"  ]]; then
-        NCCL_DEPS="apt-get install -y --allow-downgrades libnccl2=2.1.2-1+cuda${CUDA_MAJOR} libnccl-dev=2.1.2-1+cuda${CUDA_MAJOR} &&"
+        NCCL_DEPS="apt-get install -y --allow-downgrades libnccl2=2.2.13-1+cuda${CUDA_MAJOR} libnccl-dev=2.2.13-1+cuda${CUDA_MAJOR} &&"
    else
        NCCL_DEPS=""
    fi