2021-04-23 17:33:06 +02:00
|
|
|
FROM ubuntu:latest
|
|
|
|
|
|
2021-05-13 23:06:29 +02:00
|
|
|
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/*
|
2021-04-23 17:33:06 +02:00
|
|
|
RUN pip3 install torch torchvision torchaudio
|
2021-05-13 23:06:29 +02:00
|
|
|
RUN pip3 install sacred && pip3 install GitPython
|
2021-05-11 22:13:43 +02:00
|
|
|
WORKDIR /app
|
2021-04-23 17:33:06 +02:00
|
|
|
|
2021-05-10 10:39:46 +02:00
|
|
|
COPY ./../create.py ./
|
|
|
|
|
COPY ./../stats.py ./
|
|
|
|
|
COPY ./../stroke-pytorch.py ./
|
|
|
|
|
COPY ./../stroke-pytorch-eval.py ./
|
2021-04-23 17:33:06 +02:00
|
|
|
|
