diff --git a/docker/Dockerfile b/docker/Dockerfile
new file mode 100644
index 00000000..3aa86831
--- /dev/null
+++ b/docker/Dockerfile
@@ -0,0 +1,5 @@
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+
+COPY requirements.txt .
+
+RUN pip install --no-cache-dir -r requirements.txt
diff --git a/docker/requirements.txt b/docker/requirements.txt
new file mode 100644
index 00000000..69791f2d
--- /dev/null
+++ b/docker/requirements.txt
@@ -0,0 +1,7 @@
+accelerate==0.23.0
+einops==0.7.0
+diffusers==0.21.4
+transformers==4.34.0
+wandb==0.15.11
+moviepy==1.0.3
+imageio==2.31.5