FROM ubuntu:20.04

RUN apt update && apt install -y python3 && apt install -y nano

RUN apt update && apt install python3-pip -y
RUN pip3 install kaggle && pip3 install pandas && pip3 install scikit-learn && pip3 install matplotlib
RUN apt install -y curl
RUN pip3 install --user wget
RUN pip3 install torch torchvision torchaudio
RUN pip3 install sacred && pip3 install GitPython && pip3 install pymongo
RUN pip3 install dvc
RUN pip3 install 'dvc[ssh]' paramiko

WORKDIR /app

COPY ./init.py ./
COPY ./stats.py ./
COPY ./pytorch-example.py ./
COPY ./sacred-example-file.py ./
COPY ./sacred-example-mongo.py ./

RUN mkdir /.kaggle
RUN chmod -R 777 /.kaggle