ium_452627/Dockerfile
2023-05-07 18:54:27 +02:00

31 lines
728 B
Docker

# 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