From 8a781d6dbebc6b9b2d1a63712651e09d81fe579f Mon Sep 17 00:00:00 2001
From: Yuxiang Gao <yuxiang.gao@fftai.com>
Date: Sat, 21 Dec 2024 18:56:24 +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 000000000..e75eb2ba9
--- /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