FROM ubuntu:latest

RUN apt-get update && apt-get install -y python3-pip && pip3 install setuptools && pip3 install numpy && pip3 install pandas && pip3 install wget && pip3 install scikit-learn && pip3 install matplotlib  && rm -rf /var/lib/apt/lists/*
RUN pip3 install torch torchvision torchaudio 
RUN pip3 install sacred && pip3 install GitPython && pip3 install pymongo
WORKDIR /app 

COPY ./../create.py ./
COPY ./../stats.py ./
COPY ./../stroke-pytorch.py ./
COPY ./../stroke-pytorch-eval.py ./
COPY ./../sacred-pytorch2.py ./
COPY ./../sacred-pytorch1.py ./