ium_452627/Dockerfile

# Nasz obraz będzie dzidziczył z obrazu Ubuntu w wersji latest
FROM ubuntu:latest

# Instalujemy niezbędne zależności. Zwróć uwagę na flagę "-y" (assume yes)
RUN apt update && apt install -y figlet

WORKDIR /app

RUN apt-get install -y python3
RUN apt-get install -y python3-pip
RUN python3 -m pip install pandas
RUN python3 -m pip install numpy
RUN python3 -m pip install torch
RUN python3 -m pip install torchvision

COPY ./zadanie1.py ./
COPY ./Customers.csv ./
COPY ./train.py ./
#COPY ./test.py ./
#COPY ./cifar_net.pth ./

RUN chmod +r ./Customers.csv
#RUN chmod +x ./zadanie1.py
RUN chmod +x ./train.py
#RUN chmod +x ./test.py

ARG epochs=5
RUN echo $epochs

RUN python3 ./train.py $epochs
#CMD python3 ./test.py