FROM nvidia/cuda:11.0.3-base-ubuntu20.04

WORKDIR /usr/src/app

RUN apt-get update
RUN apt-get install python3-pip -y

RUN pip3 install pandas==1.5.2 \
    boto3==1.24.15 \
    fsspec==2022.10.0 \
    s3fs==2022.10.0 

COPY training.py .
RUN mkdir refdata

HEALTHCHECK NONE

CMD ["python3", "-u", "training.py"]