diff --git a/docker/Dockerfile b/docker/Dockerfile index 471c8c6..aaa4d55 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -56,11 +56,6 @@ RUN pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url h # Clone Hunyuan3D-2.1 repository RUN git clone https://github.com/Tencent-Hunyuan/Hunyuan3D-2.1.git -# Remove Chinese mirror sources and torchdiffeq from requirements.txt -RUN cd /workspace/Hunyuan3D-2.1 && \ - sed -i '/--extra-index-url https:\/\/mirrors\.cloud\.tencent\.com\/pypi\/simple\//d' requirements.txt && \ - sed -i '/--extra-index-url https:\/\/mirrors\.aliyun\.com\/pypi\/simple/d' requirements.txt - # Install Python dependencies from modified requirements.txt RUN pip install -r Hunyuan3D-2.1/requirements.txt @@ -79,6 +74,19 @@ RUN cd /workspace/Hunyuan3D-2.1/hy3dpaint/DifferentiableRenderer && \ RUN cd /workspace/Hunyuan3D-2.1 && \ wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P ckpt +# Create ckpt folder in hy3dpaint and download RealESRGAN model +RUN cd /workspace/Hunyuan3D-2.1/hy3dpaint && \ + mkdir -p ckpt && \ + wget https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth -P ckpt + +# Modify textureGenPipeline.py to update config path +RUN cd /workspace/Hunyuan3D-2.1/hy3dpaint && \ + sed -i 's/self\.multiview_cfg_path = "cfgs\/hunyuan-paint-pbr\.yaml"/self.multiview_cfg_path = "hy3dpaint\/cfgs\/hunyuan-paint-pbr.yaml"/' textureGenPipeline.py + +# Modify multiview_utils.py to update custom_pipeline path +RUN cd /workspace/Hunyuan3D-2.1/hy3dpaint/utils && \ + sed -i 's/custom_pipeline = config\.custom_pipeline/custom_pipeline = os.path.join(os.path.dirname(__file__),"..","hunyuanpaintpbr")/' multiview_utils.py + # Set working directory to the cloned repository WORKDIR /workspace/Hunyuan3D-2.1 diff --git a/docker/README.md b/docker/README.md new file mode 100644 index 0000000..bfd672d --- /dev/null +++ b/docker/README.md @@ -0,0 +1,33 @@ +# Docker setup + +This docker setup is tested on Windows 10. + +make sure you are under this directory yourworkspace/Hunyuan3D-2.1/docker + +Build docker image: + +``` +docker build -t hunyuan3d21:latest . +``` + +Run docker image at the first time: + +``` +docker run -it --name hy3d21 -p 7860:7860 --gpus all hunyuan3d21 python gradio_app.py --port 7860 +``` + +After first time: +``` +docker start -a hy3d21 +``` + +Stop the container: +``` +docker stop hy3d21 +``` + +You can find the demo link showing in terminal, such as `http://0.0.0.0:7860`, then you cuold access `http://127.0.0.1:7860` from your host machine. + +Some notes: +1. the total built time might take more than one hour. +2. the total size of the built image will be more than 70GB. \ No newline at end of file