|
|
|
@ -415,9 +415,11 @@ function gen_dockerfile() {
|
|
|
|
|
|
|
|
|
|
DOCKERFILE_GPU_ENV=""
|
|
|
|
|
DOCKERFILE_CUDNN_DSO=""
|
|
|
|
|
DOCKERFILE_CUBLAS_DSO=""
|
|
|
|
|
if [[ ${WITH_GPU:-OFF} == 'ON' ]]; then
|
|
|
|
|
DOCKERFILE_GPU_ENV="ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu:\${LD_LIBRARY_PATH}"
|
|
|
|
|
DOCKERFILE_CUDNN_DSO="RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.${CUDNN_MAJOR} /usr/lib/x86_64-linux-gnu/libcudnn.so"
|
|
|
|
|
DOCKERFILE_CUDNN_DSO="RUN ln -sf /usr/lib/x86_64-linux-gnu/libcudnn.so.${CUDNN_MAJOR} /usr/lib/x86_64-linux-gnu/libcudnn.so"
|
|
|
|
|
DOCKERFILE_CUBLAS_DSO="RUN ln -sf /usr/local/cuda/targets/x86_64-linux/lib/libcublas.so.${CUDNN_MAJOR} /usr/lib/x86_64-linux-gnu/libcublas.so"
|
|
|
|
|
fi
|
|
|
|
|
|
|
|
|
|
cat <<EOF
|
|
|
|
@ -458,6 +460,7 @@ EOF
|
|
|
|
|
${PADDLE_VERSION} && \
|
|
|
|
|
ldconfig
|
|
|
|
|
${DOCKERFILE_CUDNN_DSO}
|
|
|
|
|
${DOCKERFILE_CUBLAS_DSO}
|
|
|
|
|
${DOCKERFILE_GPU_ENV}
|
|
|
|
|
ENV NCCL_LAUNCH_MODE PARALLEL
|
|
|
|
|
EOF
|
|
|
|
|