Skip to content

Commit 1b1f0c4

Browse files
author
DvirDukhan
committed
copy only cuda libs
1 parent 9a7a3eb commit 1b1f0c4

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

Dockerfile.gpu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ WORKDIR /build
3636
COPY --from=redis /usr/local/ /usr/local/
3737

3838
COPY --from=cuda_10.2 /usr/local/cuda-10.2 /usr/local/cuda-10.2
39-
COPY --from=cuda_10.2 /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu
39+
COPY --from=cuda_10.2 /usr/lib/x86_64-linux-gnu/libcu* /usr/lib/x86_64-linux-gnu/
4040

4141
RUN echo export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-11.0/lib64:/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH > /etc/profile.d/cuda.sh
4242

@@ -84,7 +84,7 @@ RUN mkdir -p $REDIS_MODULES/
8484

8585
COPY --from=redis /usr/local/ /usr/local/
8686
COPY --from=builder /usr/local/cuda-10.2 /usr/local/cuda-10.2
87-
COPY --from=builder /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu
87+
COPY --from=builder /usr/lib/x86_64-linux-gnu/libcu* /usr/lib/x86_64-linux-gnu/
8888
COPY --from=builder /build/install-gpu/ $REDIS_MODULES/
8989

9090
COPY --from=builder /build/bin/artifacts/ /var/opt/redislabs/artifacts

Dockerfile.gpu-test

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ WORKDIR /build
2929
COPY --from=redis /usr/local/ /usr/local/
3030

3131
COPY --from=cuda_10.2 /usr/local/cuda-10.2 /usr/local/cuda-10.2
32-
COPY --from=cuda_10.2 /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu
32+
COPY --from=cuda_10.2 /usr/lib/x86_64-linux-gnu/libcu* /usr/lib/x86_64-linux-gnu/
3333

3434
RUN echo export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-11.0/lib64:/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH > /etc/profile.d/cuda.sh
3535

0 commit comments

Comments
 (0)