ium_444354/pytorch/Dockerfile

33 lines
802 B
Docker
Raw Normal View History

2022-04-17 19:33:40 +02:00
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 random
RUN pip3 install torchvision
# COPY ./script.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.py ./
# Script executed after docker run
# CMD python3 ./pytorch.py
RUN chmod o+wrx ./runTrain.sh
RUN ./runTrain.sh