ium_434766/Dockerfile
s434766 1558c8c6c4
All checks were successful
s434766-training/pipeline/head This commit looks good
lab5 .
2021-04-21 21:01:14 +02:00

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