Dockerfile: fix production image libnccl install error. (#10848)

faedee0d · Lei Wang · GitHub · e858471a · faedee0d
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/scripts/paddle_build.sh paddle/scripts/paddle_build.sh +1 -1

未找到文件。
--- a/paddle/scripts/paddle_build.sh
+++ b/paddle/scripts/paddle_build.sh
@@ -433,7 +433,7 @@ EOF
 EOF

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