Skip to content

Commit c78efe4

Browse files
authored
Merge branch 'master' into DAG_execute_command
2 parents 112649b + dc94901 commit c78efe4

File tree

4 files changed

+11
-14
lines changed

4 files changed

+11
-14
lines changed

Dockerfile.gpu

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ ARG TEST=0
1818

1919
#----------------------------------------------------------------------------------------------
2020
FROM redisfab/redis:${REDIS_VER}-${ARCH}-${OSNICK} AS redis
21-
FROM nvidia/cuda:10.2-cudnn8-devel-${OS} AS cuda_10.2
2221
FROM nvidia/cuda:${CUDA_VER}-devel-${OS} AS builder
2322

2423
ARG OSNICK
@@ -35,10 +34,7 @@ ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
3534
WORKDIR /build
3635
COPY --from=redis /usr/local/ /usr/local/
3736

38-
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/libcu* /usr/lib/x86_64-linux-gnu/
40-
41-
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
37+
RUN echo export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH > /etc/profile.d/cuda.sh
4238

4339
COPY ./opt/ opt/
4440
COPY ./tests/flow/tests_setup/test_requirements.txt tests/flow/
@@ -75,13 +71,12 @@ FROM nvidia/cuda:${CUDA_VER}-runtime-${OS}
7571
ARG OS
7672

7773
RUN if [ ! -z $(command -v apt-get) ]; then apt-get -qq update; apt-get -q install -y libgomp1; fi
78-
RUN if [ ! -z $(command -v yum) ]; then yum install -y libgomp; fi
74+
RUN if [ ! -z $(command -v yum) ]; then yum install -y libgomp; fi
7975

8076
ENV REDIS_MODULES /usr/lib/redis/modules
8177
RUN mkdir -p $REDIS_MODULES/
8278

8379
COPY --from=redis /usr/local/ /usr/local/
84-
COPY --from=builder /usr/local/cuda-10.2 /usr/local/cuda-10.2
8580
COPY --from=builder /usr/lib/x86_64-linux-gnu/libcu* /usr/lib/x86_64-linux-gnu/
8681
COPY --from=builder /build/install-gpu/ $REDIS_MODULES/
8782

Dockerfile.gpu-test

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ ARG PACK=1
1717

1818
#----------------------------------------------------------------------------------------------
1919
FROM redisfab/redis:${REDIS_VER}-${ARCH}-${OSNICK} AS redis
20-
FROM nvidia/cuda:10.2-cudnn8-devel-${OS} AS cuda_10.2
2120
FROM nvidia/cuda:${CUDA_VER}-devel-${OS} AS builder
2221

2322
SHELL ["/bin/bash", "-c"]
@@ -28,10 +27,7 @@ ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
2827
WORKDIR /build
2928
COPY --from=redis /usr/local/ /usr/local/
3029

31-
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/libcu* /usr/lib/x86_64-linux-gnu/
33-
34-
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
30+
RUN echo export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH > /etc/profile.d/cuda.sh
3531

3632
COPY ./opt/ opt/
3733
COPY ./tests/flow/tests_setup/test_requirements.txt tests/flow/tests_setup/

opt/pack.sh

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,13 @@ BINDIR=$(realpath $BINDIR)
5151
INSTALL_DIR=$(realpath $INSTALL_DIR)
5252

5353
$READIES/enable-utf8
54-
source /etc/profile.d/utf8.sh
54+
if [ -f /etc/profile.d/utf8.sh ]; then
55+
source /etc/profile.d/utf8.sh
56+
else
57+
echo export LC_ALL=C.UTF-8 >> /etc/profile.d/utf8.sh
58+
echo export LANG=C.UTF-8 >> /etc/profile.d/utf8.sh
59+
source /etc/profile.d/utf8.sh
60+
fi
5561

5662
export ARCH=$($READIES/platform --arch)
5763
export OS=$($READIES/platform --os)

0 commit comments

Comments
 (0)