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 sacred && pip3 install matplotlib && rm -rf /var/lib/apt/lists/* RUN pip3 install torch torchvision torchaudio WORKDIR /app COPY ./create.py ./ COPY ./stats.py ./ COPY ./stroke-pytorch.py ./ COPY ./stroke-pytorch-eval.py ./