File tree Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Original file line number Diff line number Diff line change 847
847
${DOCKERFILE_CUDNN_DSO}
848
848
${DOCKERFILE_CUBLAS_DSO}
849
849
${DOCKERFILE_GPU_ENV}
850
- ENV NCCL_LAUNCH_MODE PARALLEL
851
850
EOF
852
851
elif [ " $1 " == " cp36-cp36m" ]; then
853
852
cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
873
872
${DOCKERFILE_CUDNN_DSO}
874
873
${DOCKERFILE_CUBLAS_DSO}
875
874
${DOCKERFILE_GPU_ENV}
876
- ENV NCCL_LAUNCH_MODE PARALLEL
877
875
EOF
878
876
elif [ " $1 " == " cp37-cp37m" ]; then
879
877
cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
896
894
${DOCKERFILE_CUDNN_DSO}
897
895
${DOCKERFILE_CUBLAS_DSO}
898
896
${DOCKERFILE_GPU_ENV}
899
- ENV NCCL_LAUNCH_MODE PARALLEL
900
897
EOF
901
898
else
902
899
cat >> ${PADDLE_ROOT} /build/Dockerfile << EOF
912
909
${DOCKERFILE_CUDNN_DSO}
913
910
${DOCKERFILE_CUBLAS_DSO}
914
911
${DOCKERFILE_GPU_ENV}
915
- ENV NCCL_LAUNCH_MODE PARALLEL
916
912
EOF
917
913
fi
918
914
You can’t perform that action at this time.
0 commit comments