2021-04-10 13:22:11 +02:00
|
|
|
FROM ubuntu:latest
|
|
|
|
|
2021-04-17 13:35:20 +02:00
|
|
|
RUN apt update && apt install -y python3-pip && pip3 install setuptools && pip3 install numpy && pip3 install pandas && pip3 install wget && pip3 install scikit-learn && pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html &&rm -rf /var/lib/apt/lists/*
|
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-04-10 13:57:58 +02:00
|
|
|
|