Dockerfile.jetson

# BUILD redisfab/redisai:${VERSION}-jetson-${ARCH}-${OSNICK}

ARG REDIS_VER=6.0.9

# OSNICK=bionic|centos7|centos6
ARG OSNICK=bionic

# OS=ubuntu18.04|ubuntu16.04|centos7
ARG OS=ubuntu18.04

# ARCH=arm64v8|arm32v7
ARG ARCH=arm64v8

ARG CUDA_VER=10.2-cudnn7

ARG L4T_VER=r32.4.4

ARG PACK=0
ARG TEST=0

#----------------------------------------------------------------------------------------------
FROM redisfab/redis:${REDIS_VER}-${ARCH}-${OSNICK} AS redis
FROM redisfab/jetpack:4.4.1-arm64v8-l4t as builder

ARG OSNICK
ARG OS
ARG ARCH
ARG REDIS_VER
ARG CUDA_VER
ARG L4T_VER

RUN echo "Building for ${OSNICK} (${OS}) for ${ARCH} [with Redis ${REDIS_VER}]"

ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility

WORKDIR /build
COPY --from=redis /usr/local/ /usr/local/

COPY ./opt/ opt/
COPY ./tests/flow/test_requirements.txt tests/flow/

RUN FORCE=1 ./opt/readies/bin/getpy3
RUN ./opt/system-setup.py

ARG DEPS_ARGS="GPU=1 JETSON=1 WITH_PT=1 WITH_TF=1 WITH_TFLITE=0 WITH_ORT=0"
COPY ./get_deps.sh .
# RUN if [ "$DEPS_ARGS" = "" ]; then ./get_deps.sh gpu; else env $DEPS_ARGS ./get_deps.sh gpu; fi
RUN set -e ;\
    env $DEPS_ARGS ./get_deps.sh

ARG BUILD_ARGS="GPU=1 JETSON=1 WITH_TF=1 WITH_PT=1 WITH_TFLITE=0 WITH_ORT=0"

ADD ./ /build
RUN bash -c "set -e ;\
    . ./opt/readies/bin/sourced ./profile.d ;\
    make -C opt build  $BUILD_ARGS SHOW=1"

ARG PACK
ARG TEST

RUN mkdir -p bin/artifacts
RUN set -e ;\
    if [ "$PACK" = "1" ]; then make -C opt pack GPU=1; fi

RUN set -e ;\
    if [ "$TEST" = "1" ]; then \
        TEST= make -C opt test GPU=1 $BUILD_ARGS NO_LFS=1 ;\
        if [[ -d test/logs ]]; then \
            tar -C test/logs -czf bin/artifacts/test-logs-gpu.tgz . ;\
        fi ;\
    fi

#----------------------------------------------------------------------------------------------
# FROM nvidia/cuda:${CUDA_VER}-runtime-${OS}
# FROM nvidia/cuda-arm64:11.1-runtime-ubuntu18.04
# FROM nvcr.io/nvidia/l4t-base:${L4T_VER}
FROM redisfab/jetpack:4.4.1-arm64v8-l4t

ARG OS
ARG L4T_VER

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

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

COPY --from=redis /usr/local/ /usr/local/
COPY --from=builder /build/install-gpu/ $REDIS_MODULES/

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

# COPY --from=builder /usr/local/etc/redis/redis.conf /usr/local/etc/redis/

WORKDIR /data
EXPOSE 6379
# CMD ["/usr/local/bin/redis-server", "/usr/local/etc/redis/redis.conf", "--loadmodule", "/usr/lib/redis/modules/redisai.so"]
RUN rm -rf /root/.cache /usr/local/cuda/lib64/*.a /usr/local/cuda/doc /usr/local/cuda/samples
CMD ["/usr/local/bin/redis-server", "--loadmodule", "/usr/lib/redis/modules/redisai.so"]