Skip to content

Commit 6737c75

Browse files
author
Helin Wang
committed
fix gpu dockerbuild
1 parent 4951c46 commit 6737c75

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

paddle/scripts/docker/build.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,16 @@ fi
1414
DOCKERFILE_GPU_ENV=""
1515
if [[ ${WITH_GPU:-OFF} == 'ON' ]]; then
1616
DOCKERFILE_GPU_ENV="ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu:${LD_LIBRARY_PATH}"
17+
18+
# for cmake to find cudnn
19+
ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so /usr/lib/libcudnn.so
1720
fi
1821

1922
mkdir -p /paddle/build
2023
cd /paddle/build
2124

2225
# build script will not fail if *.deb does not exist
23-
rm *.deb || true
26+
rm *.deb 2>/dev/null || true
2427

2528
cmake .. \
2629
-DCMAKE_BUILD_TYPE=Release \

0 commit comments

Comments
 (0)