Ashton99 commited on
Commit
eed042a
·
verified ·
1 Parent(s): 9c5f8f4

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +18 -0
Dockerfile ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
2
+
3
+ # 1. basic tools
4
+ RUN apt-get update && apt-get install -y git ffmpeg && rm -rf /var/lib/apt/lists/*
5
+
6
+ # 2. clone Matrix-Game
7
+ RUN git clone --depth 1 https://github.com/SkyworkAI/Matrix-Game.git /workspace/Matrix-Game
8
+
9
+ # 3. install its dependencies & the package itself
10
+ RUN pip install --no-cache-dir -r /workspace/Matrix-Game/Matrix-Game-2/requirements.txt \
11
+ && pip install --no-cache-dir -e /workspace/Matrix-Game/Matrix-Game-2
12
+
13
+ # 4. gradio UI
14
+ RUN pip install --no-cache-dir gradio
15
+
16
+ # 5. copy the simple interface you already created
17
+ COPY app.py /app.py
18
+ CMD ["python", "/app.py"]