ium_434766/Dockerfile

15 lines
567 B
Docker
Raw Permalink Normal View History

2021-06-12 11:49:08 +02:00
FROM ubuntu:20.04
2021-04-10 13:22:11 +02:00
2021-05-11 22:13:02 +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/*
RUN pip3 install torch torchvision torchaudio
2021-05-13 23:23:52 +02:00
RUN pip3 install sacred && pip3 install GitPython && pip3 install pymongo
RUN pip3 install dvc
RUN pip3 install 'dvc[ssh]' paramiko
2021-04-10 13:22:11 +02:00
WORKDIR /app
COPY ./create.py ./
COPY ./stats.py ./
2021-04-17 13:35:20 +02:00
COPY ./stroke-pytorch.py ./
2021-05-07 21:30:35 +02:00
COPY ./stroke-pytorch-eval.py ./
COPY ./train-dvc.py ./