Dockerfile.cuda10_1

FROM nvcr.io/nvidia/cuda:10.1-runtime

RUN apt-get update && apt-get install -y openjdk-8-jre-headless git python3 python3-pip && python3 -m pip install pipenv

ENV LC_ALL=C.UTF-8
ENV LANG=C.UTF-8

WORKDIR /workspace

RUN git clone -b container-test https://github.com/willb/telco-churn-augmentation .

RUN grep -v altair < Pipfile > Pipfile.tmp && mv Pipfile.tmp Pipfile

RUN pipenv install

ADD https://repo1.maven.org/maven2/ai/rapids/cudf/0.17/cudf-0.17-cuda10-1.jar /workspace
ADD https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/0.3.0/rapids-4-spark_2.12-0.3.0.jar /workspace

COPY start.sh /workspace

RUN chmod 755 /workspace/start.sh

CMD ["./start.sh"]