From b8c896814afc0157ecc83d712ee697c0a04a06d8 Mon Sep 17 00:00:00 2001
From: zhaoyongke <zhaoyongke@yeah.net>
Date: Wed, 9 Nov 2022 11:32:51 +0000
Subject: [PATCH] Add dockerfile with NGC base image, supports H100 and RTX
 4090 arch

---
 Dockerfile | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 Dockerfile

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000000..4ff23933db
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,23 @@
+FROM nvcr.io/nvidia/pytorch:22.10-py3
+
+WORKDIR /research
+
+ENV HOME /research
+
+RUN pip install omegaconf==2.1.1 diffusers invisible-watermark opencv-python-headless==4.1.2.30 einops==0.3.0 pytorch-lightning==1.4.2  torchmetrics==0.6.0 kornia==0.6 transformers==4.19.2
+
+# Mount data into the docker
+ADD . /research/stable-diffusion
+
+
+WORKDIR /research/stable-diffusion
+ENV PYTHONPATH /research/stable-diffusion
+
+RUN mkdir src && cd src && \
+    git clone https://github.com/CompVis/taming-transformers.git && cd taming-transformers && \
+    pip install -e . && cd ../
+RUN git clone https://github.com/openai/CLIP.git && cd CLIP && \
+    pip install -e . && cd ../../
+
+ENTRYPOINT ["/bin/bash"]
+