31 lines
728 B
Docker
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 |