Skip to content

Commit e68d196

Browse files
committed
update cudnn version to cuda version
1 parent 1087bb0 commit e68d196

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddle/scripts/paddle_build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -419,7 +419,7 @@ function gen_dockerfile() {
419419
if [[ ${WITH_GPU:-OFF} == 'ON' ]]; then
420420
DOCKERFILE_GPU_ENV="ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu:\${LD_LIBRARY_PATH}"
421421
DOCKERFILE_CUDNN_DSO="RUN ln -sf /usr/lib/x86_64-linux-gnu/libcudnn.so.${CUDNN_MAJOR} /usr/lib/x86_64-linux-gnu/libcudnn.so"
422-
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"
422+
DOCKERFILE_CUBLAS_DSO="RUN ln -sf /usr/local/cuda/targets/x86_64-linux/lib/libcublas.so.${CUDA_MAJOR} /usr/lib/x86_64-linux-gnu/libcublas.so"
423423
fi
424424

425425
cat <<EOF

0 commit comments

Comments
 (0)