11 lines
355 B
Docker
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 ./
|
||
|
|
|