35 lines
842 B
Plaintext
35 lines
842 B
Plaintext
|
FROM ubuntu:latest
|
||
|
|
# Install required dependencies
|
||
|
|
RUN apt update
|
||
|
|
RUN apt-get update
|
||
|
|
RUN apt install -y figlet
|
||
|
|
RUN export PATH=”$PATH:/usr/local/bin/python”
|
||
|
|
RUN apt install python3-pip -y
|
||
|
|
RUN apt install unzip -y
|
||
|
|
RUN pip3 install --upgrade pip
|
||
|
|
RUN pip3 install --upgrade Pillow
|
||
|
|
RUN pip3 install pandas
|
||
|
|
RUN pip3 install scikit-learn
|
||
|
|
RUN pip3 install matplotlib
|
||
|
|
RUN pip3 install jovian
|
||
|
|
RUN pip3 install torch
|
||
|
|
RUN pip3 install seaborn
|
||
|
|
RUN pip3 install torchvision
|
||
|
|
|
||
|
|
|
||
|
|
# COPY ./script.sh ./
|
||
|
|
COPY ./runTrain.sh ./runTrain.sh
|
||
|
|
|
||
|
|
# RUN chmod a+x ./script.sh
|
||
|
|
# Create app directory in image
|
||
|
|
WORKDIR /app
|
||
|
|
COPY ./winequality-red.csv ./
|
||
|
|
# Copy init dataset script to /app directory in image
|
||
|
|
COPY ./pytorch/pytorch.py ./
|
||
|
|
|
||
|
|
# Script executed after docker run
|
||
|
|
# CMD python3 ./pytorch.py
|
||
|
|
COPY ./runTrain.sh ./
|
||
|
|
RUN chmod a+x ./runTrain.sh
|
||
|
|
RUN ./runTrain.sh
|