diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..e75eb2b
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,23 @@
+FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn9-devel
+ARG DEBIAN_FRONTEND=noninteractive
+
+# RUN sed -i 's/archive.ubuntu.com/mirrors.tuna.tsinghua.edu.cn/g' /etc/apt/sources.list
+
+RUN apt update && apt install -y \
+    git \
+    vim \
+    wget \
+    curl \
+    libgl1-mesa-glx \
+    libxrender1 \
+    libglib2.0-0 \
+    ffmpeg \
+    libgtk2.0-dev \
+    pkg-config \
+    && apt clean \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY . /workspace
+WORKDIR /workspace
+
+RUN pip install --no-cache-dir PyOpenGL==3.1.1a1 && pip install -e . --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple
\ No newline at end of file