Initial Commit

Dockerfile

@@ -1,13 +1,8 @@
-FROM pytorch/pytorch
-
-WORKDIR /app
-COPY ./requirements.txt /app/
-
-RUN pip install -r requirements.txt
-RUN apt-get -y update
-RUN apt-get -y install git
-RUN apt-get install ffmpeg libsm6 libxext6  -y
-
-COPY . /app/
-RUN git config --global --add safe.directory /app
-CMD ./train.sh
+FROM ubuntu:20.04
+ADD requirements.txt /requirements.txt
+ARG DEBIAN_FRONTEND=noninteractive
+RUN apt-get update
+RUN apt-get install ffmpeg libsm6 libxext6  git -y
+RUN apt-get install -y libglib2.0-0
+RUN apt-get -y install python3-pip
+RUN pip install -r /requirements.txt