davidvgilmore commited on
Commit
06605e4
Β·
1 Parent(s): 148d968

Update configuration to use Docker with CUDA

Browse files
Files changed (2) hide show
  1. Dockerfile +32 -0
  2. README.md +2 -1
Dockerfile ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvidia/cuda:12.3.1-devel-ubuntu22.04
2
+
3
+ # Install Python and basic utilities
4
+ RUN apt-get update && apt-get install -y \
5
+ python3.10 \
6
+ python3-pip \
7
+ git \
8
+ wget \
9
+ && rm -rf /var/lib/apt/lists/*
10
+
11
+ # Set working directory
12
+ WORKDIR /app
13
+
14
+ # Copy application files
15
+ COPY . .
16
+
17
+ # Install Python dependencies
18
+ RUN pip3 install --no-cache-dir -r requirements.txt
19
+
20
+ # Install build dependencies
21
+ RUN pip3 install --no-cache-dir ninja pybind11
22
+
23
+ # Compile custom components
24
+ RUN cd /app/hy3dgen/texgen/custom_rasterizer && python3 setup.py install
25
+ RUN cd /app/hy3dgen/texgen/differentiable_renderer && bash compile_mesh_painter.sh
26
+
27
+ # Set environment variables
28
+ ENV NVIDIA_VISIBLE_DEVICES=all
29
+ ENV PYTHONPATH=/app
30
+
31
+ # Run the application
32
+ CMD ["python3", "hg_app.py"]
README.md CHANGED
@@ -3,7 +3,7 @@ title: Hunyuan3D-2 Clone
3
  emoji: 🌍
4
  colorFrom: purple
5
  colorTo: red
6
- sdk: gradio
7
  sdk_version: 4.44.1
8
  app_file: hg_app.py
9
  pinned: false
@@ -15,3 +15,4 @@ models:
15
  # Hunyuan3D-2 Clone
16
 
17
  This is a clone of the original Hunyuan3D-2 Space by Tencent.
 
 
3
  emoji: 🌍
4
  colorFrom: purple
5
  colorTo: red
6
+ sdk: docker
7
  sdk_version: 4.44.1
8
  app_file: hg_app.py
9
  pinned: false
 
15
  # Hunyuan3D-2 Clone
16
 
17
  This is a clone of the original Hunyuan3D-2 Space by Tencent.
18
+ Hardware: T4 GPU