From 8a9c1de43cd070394331782d94e10b41bf1a6f23 Mon Sep 17 00:00:00 2001 From: Jeffreytsai1004 Date: Wed, 19 Feb 2025 03:41:30 +0800 Subject: [PATCH] Update install.bat --- install.bat | 123 ++++++++++++++++++++-------------------------------- 1 file changed, 46 insertions(+), 77 deletions(-) diff --git a/install.bat b/install.bat index 3243d67..1cdcbc2 100644 --- a/install.bat +++ b/install.bat @@ -1,6 +1,4 @@ -@echo off - -@CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\" +@CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\" @CALL SET PROJECT_DIR=%~dp0 @CALL SET MICROMAMBA_EXE=%PROJECT_DIR%\micromamba.exe @CALL SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\" @@ -9,110 +7,81 @@ @CALL SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat" @CALL SET PATH=%CUDA_HOME%\bin;%PROJECT_DIR%\condabin;%PATH% -IF NOT EXIST "%CUDA_HOME%" ( - echo ERROR: CUDA_HOME path does not exist: %CUDA_HOME% - exit /b 1 -) -IF NOT EXIST "%BLENDER_DIR%" ( - echo ERROR: BLENDER_DIR path does not exist: %BLENDER_DIR% - exit /b 1 -) -IF NOT EXIST "%VS_DIR%" ( - echo ERROR: VS_DIR path does not exist: %VS_DIR% - exit /b 1 -) -IF NOT EXIST "%MICROMAMBA_EXE%" ( - echo ERROR: micromamba not found at %MICROMAMBA_EXE% - echo Please install micromamba from https://mamba.readthedocs.io/en/latest/installation.html - exit /b 1 -) - {{ Gaussian Splatting Hair 环境 }} @CALL "%~dp0micromamba.exe" create -n gaussian_splatting_hair python=3.9 -c pytorch -c nvidia -c conda-forge -c anaconda -c fvcore -c iopath -c bottler -c nvidia -r "%~dp0\" -y -@CALL condabin\micromamba.bat activate gaussian_splatting_hair +@CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair @CALL python -m pip install pip==23.3.1 @CALL python -m pip install gdown==5.2.0 @CALL python -m pip install -r requirements.txt -@CALL cd %PROJECT_DIR%\ext -@CALL git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose --depth 1 -@CALL cd %PROJECT_DIR%\ext\openpose -@CALL git submodule update --init --recursive --remote -@CALL cd %PROJECT_DIR%\ext -@CALL git clone https://github.com/hustvl/Matte-Anything -@CALL cd %PROJECT_DIR%\ext\Matte-Anything -@CALL git clone https://github.com/IDEA-Research/GroundingDINO.git -@CALL cd %PROJECT_DIR%\ext -@CALL git clone git@github.com:egorzakharov/NeuralHaircut.git --recursive -@CALL cd %PROJECT_DIR%\ext -@CALL git clone https://github.com/facebookresearch/pytorch3d -@CALL cd %PROJECT_DIR%\ext\pytorch3d -@CALL git checkout 2f11ddc5ee7d6bd56f2fb6744a16776fab6536f7 -@CALL cd %PROJECT_DIR%\ext -@CALL git clone https://github.com/camenduru/simple-knn -@CALL cd %PROJECT_DIR%\ext\diff_gaussian_rasterization_hair\third_party -@CALL git clone https://github.com/g-truc/glm -@CALL cd %PROJECT_DIR%\ext\diff_gaussian_rasterization_hair\third_party\glm -@CALL git checkout 5c46b9c07008ae65cb81ab79cd677ecc1934b903 -@CALL cd %PROJECT_DIR%\ext -@CALL git clone --recursive https://github.com/NVIDIAGameWorks/kaolin -@CALL cd %PROJECT_DIR%\ext\kaolin -@CALL git checkout v0.15.0 -@CALL cd %PROJECT_DIR%\ext -@CALL git clone https://github.com/SSL92/hyperIQA %PROJECT_DIR%\ext\hyperIQA +@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose --depth 1 +@CALL cd %PROJECT_DIR%\ext\openpose && git submodule update --init --recursive --remote +@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/hustvl/Matte-Anything +@CALL cd %PROJECT_DIR%\ext\Matte-Anything && git clone https://github.com/IDEA-Research/GroundingDINO.git +@CALL cd %PROJECT_DIR%\ext && git clone git@github.com:egorzakharov/NeuralHaircut.git --recursive +@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/facebookresearch/pytorch3d +@CALL cd %PROJECT_DIR%\ext\pytorch3d && git checkout 2f11ddc5ee7d6bd56f2fb6744a16776fab6536f7 +@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/camenduru/simple-knn +@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/g-truc/glm +@CALL cd %PROJECT_DIR%\ext\diff_gaussian_rasterization_hair\third_party\glm && git checkout 5c46b9c07008ae65cb81ab79cd677ecc1934b903 +@CALL cd %PROJECT_DIR%\ext && git clone --recursive https://github.com/NVIDIAGameWorks/kaolin +@CALL cd %PROJECT_DIR%\ext\kaolin && git checkout v0.15.0 +@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/SSL92/hyperIQA -@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior %PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior -@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\PIXIE %PROJECT_DIR%\ext\NeuralHaircut\PIXIE +@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\ %PROJECT_DIR%\ext\NeuralHaircut\ @CALL copy %PROJECT_DIR%\resource\hyperIQA\pretrained %PROJECT_DIR%\ext\hyperIQA\pretrained @CALL copy %PROJECT_DIR%\resource\openpose %PROJECT_DIR%\ext\openpose @CALL cd %PROJECT_DIR% -@CALL condabin\micromamba.bat deactivate gaussian_splatting_hair +@CALL "%~dp0condabin\micromamba.bat" deactivate gaussian_splatting_hair {{ Matte Anything 环境 }} -@CALL "%~dp0micromamba.exe" create -n matte_anything python=3.9 pytorch=2.0.0 pytorch-cuda=11.8 torchvision tensorboard timm=0.5.4 opencv=4.5.3 mkl=2024.0 setuptools=58.2.0 easydict wget scikit-image gradio=3.46.1 fairscale -c pytorch -c nvidia -c conda-forge -r "%~dp0\" -y -@CALL condabin\micromamba.bat activate matte_anything -@CALL mkdir %PROJECT_DIR%\ext\Matte-Anything +@CALL "%~dp0micromamba.exe" create -n matte_anything ^ + python=3.9 pytorch=2.0.0 pytorch-cuda=11.8 torchvision tensorboard timm=0.5.4 opencv=4.5.3 ^ + mkl=2024.0 setuptools=58.2.0 easydict wget scikit-image gradio=3.46.1 fairscale ^ + -c pytorch -c nvidia -c conda-forge -r "%~dp0\" -y +@CALL "%~dp0condabin\micromamba.bat" activate matte_anything @CALL cd %PROJECT_DIR%\ext\Matte-Anything -@CALL git clone git@github.com:facebookresearch/segment-anything.git -@CALL cd segment-anything -@CALL pip install -e . +@CALL git clone git@github.com:facebookresearch/segment-anything.git %PROJECT_DIR%\ext\Matte-Anything\segment-anything +@CALL cd %PROJECT_DIR%\ext\Matte-Anything\segment-anything && pip install -e . @CALL cd %PROJECT_DIR%\ext\Matte-Anything -@CALL git clone https://github.com/conansherry/detectron2 -@CALL cd detectron2 -@CALL pip install -e . -@CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO -@CALL pip install -e . -@CALL mkdir %PROJECT_DIR%\ext\Matte-Anything\pretrained -@CALL cd %PROJECT_DIR%\ext\Matte-Anything\pretrained -@CALL copy %PROJECT_DIR%\resource\Matte-Anything\pretrained %PROJECT_DIR%\ext\Matte-Anything\pretrained +@CALL git clone https://github.com/conansherry/detectron2 %PROJECT_DIR%\ext\Matte-Anything\detectron2 +@CALL cd %PROJECT_DIR%\ext\Matte-Anything\detectron2 && pip install -e . +@CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO && pip install -e . +@CALL pip install supervision==0.22.0 +@CALL cd %PROJECT_DIR%\ext\Matte-Anything && mkdir pretrained +@CALL copy %PROJECT_DIR%\resource\Matte-Anything\ %PROJECT_DIR%\ext\Matte-Anything\ @CALL cd %PROJECT_DIR% -@CALL condabin\micromamba.bat deactivate matte_anything +@CALL "%~dp0condabin\micromamba.bat" deactivate matte_anything {{ OpenPose 环境 }} -@CALL "%~dp0micromamba.exe" create -n openpose cmake=3.20 -c conda-forge -r "%~dp0\" -y -@CALL condabin\micromamba.bat activate openpose -@CALL cd %PROJECT_DIR%\ext\openpose -@CALL git submodule update --init --recursive --remote +@CALL "%~dp0condabin\micromamba.bat" deactivate matte_anything +@CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair @CALL copy %PROJECT_DIR%\resource\openpose %PROJECT_DIR%\ext\openpose +@CALL "%~dp0condabin\micromamba.bat" deactivate gaussian_splatting_hair +@CALL git submodule update --init --recursive --remote +@CALL "%~dp0condabin\micromamba.bat" deactivate openpose +@CALL "%~dp0micromamba.exe" create -n openpose cmake=3.20 -c conda-forge -r "%~dp0\" -y +@CALL "%~dp0condabin\micromamba.bat" activate openpose +@CALL cd %PROJECT_DIR%\ext\openpose @CALL mkdir build @CALL cd build @CALL %VS_VCVARS% @CALL cmake .. -G "Visual Studio 17 2022" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off @CALL cmake --build . --config Release @CALL cd %PROJECT_DIR% -@CALL condabin\micromamba.bat deactivate openpose +@CALL "%~dp0condabin\micromamba.bat" deactivate openpose {{ PIXIE 环境 }} -@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.8 pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 fvcore pytorch3d==0.7.5 kornia matplotlib -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y -@CALL condabin\micromamba.bat activate pixie-env -@CALL cd %PROJECT_DIR%\ext -@CALL git clone https://github.com/yfeng95/PIXIE +@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.8 pytorch==2.0.0 torchvision torchaudio ^ + pytorch-cuda=11.8 fvcore pytorch3d==0.7.5 kornia matplotlib ^ + -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y +@CALL "%~dp0condabin\micromamba.bat" activate pixie-env @CALL cd %PROJECT_DIR%\ext\PIXIE -@CALL chmod +x fetch_model.sh && ./fetch_model.sh +@CALL copy %PROJECT_DIR%\resource\PIXIE\pixie_data %PROJECT_DIR%\ext\PIXIE\data @CALL pip install pyyaml==5.4.1 @CALL pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59 @CALL cd %PROJECT_DIR% -@CALL condabin\micromamba.bat deactivate pixie-env +@CALL "%~dp0condabin\micromamba.bat" deactivate pixie-env