Add dockerfile and update pytorch-lightning

Dockerfile

@@ -0,0 +1,22 @@
+FROM nvidia/cuda:11.0-base
+RUN apt update
+RUN apt install python3 python3-dev python3-pip wget build-essential git -y
+RUN python3 -m pip install --upgrade pip
+RUN apt-get clean -y
+RUN rm -rf /var/lib/apt/lists/*
+
+# Install miniconda
+ENV CONDA_DIR /opt/conda
+RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
+     /bin/bash ~/miniconda.sh -b -p /opt/conda
+
+# Put conda in path so we can use conda activate
+ENV PATH=$CONDA_DIR/bin:$PATH
+
+COPY . /diffusion
+WORKDIR /diffusion
+RUN conda env create -f environment.yaml
+RUN conda init bash
+RUN /bin/bash -c "source /root/.bashrc"
+SHELL ["conda", "run", "-n", "ldm", "/bin/bash", "-c"]
+ENTRYPOINT [ "conda", "run", "--no-capture-output", "-n", "ldm"]

environment.yaml

@@ -15,7 +15,7 @@ dependencies:
     - pudb==2019.2
     - imageio==2.9.0
     - imageio-ffmpeg==0.4.2
-    - pytorch-lightning==1.4.2
+    - pytorch-lightning==1.5
     - omegaconf==2.1.1
     - test-tube>=0.7.5
     - streamlit>=0.73.1