Update install.bat

This commit is contained in:
Jeffreytsai1004 2025-02-19 03:41:30 +08:00
parent f912fd2bf7
commit 8a9c1de43c

View File

@ -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 PROJECT_DIR=%~dp0
@CALL SET MICROMAMBA_EXE=%PROJECT_DIR%\micromamba.exe @CALL SET MICROMAMBA_EXE=%PROJECT_DIR%\micromamba.exe
@CALL SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\" @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 VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat"
@CALL SET PATH=%CUDA_HOME%\bin;%PROJECT_DIR%\condabin;%PATH% @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 环境 }} {{ 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 "%~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 pip==23.3.1
@CALL python -m pip install gdown==5.2.0 @CALL python -m pip install gdown==5.2.0
@CALL python -m pip install -r requirements.txt @CALL python -m pip install -r requirements.txt
@CALL cd %PROJECT_DIR%\ext @CALL cd %PROJECT_DIR%\ext && git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose --depth 1
@CALL 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\openpose @CALL cd %PROJECT_DIR%\ext && git clone https://github.com/hustvl/Matte-Anything
@CALL git submodule update --init --recursive --remote @CALL cd %PROJECT_DIR%\ext\Matte-Anything && git clone https://github.com/IDEA-Research/GroundingDINO.git
@CALL cd %PROJECT_DIR%\ext @CALL cd %PROJECT_DIR%\ext && git clone git@github.com:egorzakharov/NeuralHaircut.git --recursive
@CALL git clone https://github.com/hustvl/Matte-Anything @CALL cd %PROJECT_DIR%\ext && git clone https://github.com/facebookresearch/pytorch3d
@CALL cd %PROJECT_DIR%\ext\Matte-Anything @CALL cd %PROJECT_DIR%\ext\pytorch3d && git checkout 2f11ddc5ee7d6bd56f2fb6744a16776fab6536f7
@CALL git clone https://github.com/IDEA-Research/GroundingDINO.git @CALL cd %PROJECT_DIR%\ext && git clone https://github.com/camenduru/simple-knn
@CALL cd %PROJECT_DIR%\ext @CALL cd %PROJECT_DIR%\ext && git clone https://github.com/g-truc/glm
@CALL git clone git@github.com:egorzakharov/NeuralHaircut.git --recursive @CALL cd %PROJECT_DIR%\ext\diff_gaussian_rasterization_hair\third_party\glm && git checkout 5c46b9c07008ae65cb81ab79cd677ecc1934b903
@CALL cd %PROJECT_DIR%\ext @CALL cd %PROJECT_DIR%\ext && git clone --recursive https://github.com/NVIDIAGameWorks/kaolin
@CALL git clone https://github.com/facebookresearch/pytorch3d @CALL cd %PROJECT_DIR%\ext\kaolin && git checkout v0.15.0
@CALL cd %PROJECT_DIR%\ext\pytorch3d @CALL cd %PROJECT_DIR%\ext && git clone https://github.com/SSL92/hyperIQA
@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 copy %PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior %PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior @CALL copy %PROJECT_DIR%\resource\NeuralHaircut\ %PROJECT_DIR%\ext\NeuralHaircut\
@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\PIXIE %PROJECT_DIR%\ext\NeuralHaircut\PIXIE
@CALL copy %PROJECT_DIR%\resource\hyperIQA\pretrained %PROJECT_DIR%\ext\hyperIQA\pretrained @CALL copy %PROJECT_DIR%\resource\hyperIQA\pretrained %PROJECT_DIR%\ext\hyperIQA\pretrained
@CALL copy %PROJECT_DIR%\resource\openpose %PROJECT_DIR%\ext\openpose @CALL copy %PROJECT_DIR%\resource\openpose %PROJECT_DIR%\ext\openpose
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
@CALL condabin\micromamba.bat deactivate gaussian_splatting_hair @CALL "%~dp0condabin\micromamba.bat" deactivate gaussian_splatting_hair
{{ Matte Anything 环境 }} {{ 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 "%~dp0micromamba.exe" create -n matte_anything ^
@CALL condabin\micromamba.bat activate matte_anything python=3.9 pytorch=2.0.0 pytorch-cuda=11.8 torchvision tensorboard timm=0.5.4 opencv=4.5.3 ^
@CALL mkdir %PROJECT_DIR%\ext\Matte-Anything 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 cd %PROJECT_DIR%\ext\Matte-Anything
@CALL git clone git@github.com:facebookresearch/segment-anything.git @CALL git clone git@github.com:facebookresearch/segment-anything.git %PROJECT_DIR%\ext\Matte-Anything\segment-anything
@CALL cd segment-anything @CALL cd %PROJECT_DIR%\ext\Matte-Anything\segment-anything && pip install -e .
@CALL pip install -e .
@CALL cd %PROJECT_DIR%\ext\Matte-Anything @CALL cd %PROJECT_DIR%\ext\Matte-Anything
@CALL git clone https://github.com/conansherry/detectron2 @CALL git clone https://github.com/conansherry/detectron2 %PROJECT_DIR%\ext\Matte-Anything\detectron2
@CALL cd detectron2 @CALL cd %PROJECT_DIR%\ext\Matte-Anything\detectron2 && pip install -e .
@CALL pip install -e . @CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO && pip install -e .
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO @CALL pip install supervision==0.22.0
@CALL pip install -e . @CALL cd %PROJECT_DIR%\ext\Matte-Anything && mkdir pretrained
@CALL mkdir %PROJECT_DIR%\ext\Matte-Anything\pretrained @CALL copy %PROJECT_DIR%\resource\Matte-Anything\ %PROJECT_DIR%\ext\Matte-Anything\
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\pretrained
@CALL copy %PROJECT_DIR%\resource\Matte-Anything\pretrained %PROJECT_DIR%\ext\Matte-Anything\pretrained
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
@CALL condabin\micromamba.bat deactivate matte_anything @CALL "%~dp0condabin\micromamba.bat" deactivate matte_anything
{{ OpenPose 环境 }} {{ OpenPose 环境 }}
@CALL "%~dp0micromamba.exe" create -n openpose cmake=3.20 -c conda-forge -r "%~dp0\" -y @CALL "%~dp0condabin\micromamba.bat" deactivate matte_anything
@CALL condabin\micromamba.bat activate openpose @CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair
@CALL cd %PROJECT_DIR%\ext\openpose
@CALL git submodule update --init --recursive --remote
@CALL copy %PROJECT_DIR%\resource\openpose %PROJECT_DIR%\ext\openpose @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 mkdir build
@CALL cd build @CALL cd build
@CALL %VS_VCVARS% @CALL %VS_VCVARS%
@CALL cmake .. -G "Visual Studio 17 2022" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off @CALL cmake .. -G "Visual Studio 17 2022" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off
@CALL cmake --build . --config Release @CALL cmake --build . --config Release
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
@CALL condabin\micromamba.bat deactivate openpose @CALL "%~dp0condabin\micromamba.bat" deactivate openpose
{{ PIXIE 环境 }} {{ 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 "%~dp0micromamba.exe" create -n pixie-env python=3.8 pytorch==2.0.0 torchvision torchaudio ^
@CALL condabin\micromamba.bat activate pixie-env pytorch-cuda=11.8 fvcore pytorch3d==0.7.5 kornia matplotlib ^
@CALL cd %PROJECT_DIR%\ext -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y
@CALL git clone https://github.com/yfeng95/PIXIE @CALL "%~dp0condabin\micromamba.bat" activate pixie-env
@CALL cd %PROJECT_DIR%\ext\PIXIE @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 pyyaml==5.4.1
@CALL pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59 @CALL pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
@CALL condabin\micromamba.bat deactivate pixie-env @CALL "%~dp0condabin\micromamba.bat" deactivate pixie-env