From 057ebdb197585deefe1a3e4a44a768db0de886ca Mon Sep 17 00:00:00 2001
From: Yuxiang Gao <yuxiang.gao@fftai.com>
Date: Sat, 21 Dec 2024 19:27:01 +0800
Subject: [PATCH] Add minimal Dockerfile

Signed-off-by: Yuxiang Gao <yuxiang.gao@fftai.com>
---
 Dockerfile | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 Dockerfile

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..88f44e5
--- /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.6 && pip install -e . --no-cache-dir
\ No newline at end of file