dockerizing

Dockerfile

@@ -0,0 +1,42 @@
+FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+WORKDIR /opt/app
+
+# System deps
+RUN apt-get update && apt-get install -y \
+    git \
+    python3 \
+    python3-pip \
+    python3-dev \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN ln -s /usr/bin/python3 /usr/bin/python
+
+# Upgrade pip
+RUN pip install --upgrade pip setuptools wheel
+
+# ---- Torch FIRST (critical) ----
+RUN pip install torch torchvision torchaudio \
+    --index-url https://download.pytorch.org/whl/cu121
+
+# ---- Clone pinned Hunyuan3D ----
+ARG HY3D_COMMIT=e2df0625bda7efd5f1daba8f3f2a4cb3d9ac85f8
+RUN git clone https://github.com/tencent/Hunyuan3D-2.git && \
+    cd Hunyuan3D-2 && git checkout ${HY3D_COMMIT}
+
+WORKDIR /opt/app/Hunyuan3D-2
+
+# ---- Install root package ----
+RUN pip install -e .
+
+# ---- Build CUDA extensions ----
+RUN pip install ./hy3dgen/texgen/custom_rasterizer/ops
+RUN pip install ./hy3dgen/texgen/differentiable_renderer/ops
+
+# ---- Copy HF handler last (better caching) ----
+WORKDIR /opt/app
+COPY handler.py .
+
+ENV PYTHONPATH=/opt/app/Hunyuan3D-2

requirements.txt

@@ -8,7 +8,6 @@ numpy
 transformers
 accelerate
 bitsandbytes
-xformers
 protobuf
 sentencepiece
 hf_xet
@@ -20,6 +19,6 @@ pymeshlab
 
 #pip install -e hy3dgen@git+https://github.com/tencent/Hunyuan3D-2/
 
--e git+https://hy3dgen@github.com/tencent/Hunyuan3D-2.git#egg=hy3dgen
-git+https://hy3dgen@github.com/tencent/Hunyuan3D-2.git#subdirectory=hy3dgen/texgen/custom_rasterizer#egg=custom_rasterizer
-git+https://hy3dgen@github.com/tencent/Hunyuan3D-2.git#subdirectory=hy3dgen/texgen/differentiable_renderer#egg=differentiable_renderer
+#-e git+https://hy3dgen@github.com/tencent/Hunyuan3D-2.git#egg=hy3dgen
+#git+https://hy3dgen@github.com/tencent/Hunyuan3D-2.git#subdirectory=hy3dgen/texgen/custom_rasterizer#egg=custom_rasterizer
+#git+https://hy3dgen@github.com/tencent/Hunyuan3D-2.git#subdirectory=hy3dgen/texgen/differentiable_renderer#egg=differentiable_renderer