ium_434766/pytorch-training-eval/Dockerfile
s434766 60ba06fd0f
Some checks failed
s434766-training/pipeline/head There was a failure building this commit
s5
2021-05-10 10:39:46 +02:00

12 lines
427 B
Docker

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
WORKDIR /app
COPY ./../create.py ./
COPY ./../stats.py ./
COPY ./../stroke-pytorch.py ./
COPY ./../stroke-pytorch-eval.py ./