This commit is contained in:
Jeffreytsai1004 2025-02-23 19:11:03 +08:00
parent a1f9230210
commit 849b60e602

View File

@ -11,7 +11,7 @@ REM Set environment variables
@CALL SET PROJECT_DIR="%~dp0" @CALL SET PROJECT_DIR="%~dp0"
@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"
@CALL SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6" @CALL SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6"
@CALL SET VS_DIR="D:\Program Files\Microsoft Visual Studio\2022\Community" @CALL SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools"
@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%
@ -22,8 +22,8 @@ REM Create gaussian_splatting_hair environment
@CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair @CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
@CALL python -m pip install --upgrade pip @CALL python -m pip install --upgrade pip
@CALL python -m pip install torch==2.1.1+cu118 torchvision torchaudio pytorch-cuda --index-url https://download.pytorch.org/whl/cu118 --force-reinstall --no-cache-dir @CALL python -m pip install torch==2.1.1+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --force-reinstall --no-cache-dir
@CALL python -m pip install pytorch-cuda cmake pyhocon icecream einops accelerate jsonmerge easydict iopath tensorboardx scikit-image gdown colmap @CALL python -m pip install cmake pyhocon icecream einops accelerate jsonmerge easydict iopath tensorboardx scikit-image gdown colmap --force-reinstall --no-cache-dir
@CALL python -m pip install pysdf clean-fid face-alignment clip torchdiffeq torchsde resize-right @CALL python -m pip install pysdf clean-fid face-alignment clip torchdiffeq torchsde resize-right
@ -69,8 +69,8 @@ REM Create matte_anything environment
@CALL "%~dp0condabin\micromamba.bat" activate matte_anything @CALL "%~dp0condabin\micromamba.bat" activate matte_anything
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\ @CALL cd %PROJECT_DIR%\ext\Matte-Anything\
@CALL python -m pip install --upgrade pip @CALL python -m pip install --upgrade pip
@CALL python -m pip install torch==2.1.1+cu118 torchvision torchaudio pytorch-cuda --index-url https://download.pytorch.org/whl/cu118 --force-reinstall --no-cache-dir @CALL python -m pip install torch==2.1.1+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --force-reinstall --no-cache-dir
@CALL python -m pip install tensorboard timm opencv mkl setuptools easydict scikit-image gradio @CALL python -m pip install mkl easydict wget scikit-image gradio fairscale pyside2 future pillow setuptools future --force-reinstall --no-cache-dir
@CALL git clone https://github.com/facebookresearch/segment-anything.git %PROJECT_DIR%\ext\Matte-Anything\segment-anything @CALL git clone https://github.com/facebookresearch/segment-anything.git %PROJECT_DIR%\ext\Matte-Anything\segment-anything
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\segment-anything @CALL cd %PROJECT_DIR%\ext\Matte-Anything\segment-anything
@CALL python -m pip install -e . @CALL python -m pip install -e .
@ -92,15 +92,17 @@ REM Create openpose environment
@CALL cd %PROJECT_DIR%\ext\openpose @CALL cd %PROJECT_DIR%\ext\openpose
@CALL python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV @CALL python -m gdown 1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV
@CALL tar -xvzf models.tar.gz @CALL tar -xvzf models.tar.gz
@CALL "%~dp0condabin\micromamba.bat" deactivate
@CALL git submodule update --init --recursive --remote @CALL git submodule update --init --recursive --remote
@CALL "%~dp0condabin\micromamba.exe" create -n openpose cmake=3.20 -c conda-forge -r "%~dp0\" -y @CALL cd %PROJECT_DIR%
@CALL "%~dp0micromamba.exe" create -y -n openpose cmake=3.20 -c conda-forge -r "%~dp0\"
@CALL "%~dp0condabin\micromamba.bat" activate openpose @CALL "%~dp0condabin\micromamba.bat" activate openpose
@CALL cd %PROJECT_DIR%\ext\openpose @CALL cd %PROJECT_DIR%\ext\openpose
@CALL rmdir /s /q build 2>nul
@CALL mkdir build @CALL mkdir build
@CALL cd build @CALL cd build
@CALL CALL %VS_VCVARS% @CALL set VS_VCVARS_FIXED="%VS_VCVARS:"=%"
@CALL cmake .. -G "Visual Studio 17 2022" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off @CALL %VS_VCVARS_FIXED%
@CALL cmake .. -G "Visual Studio 16 2019" -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 "%~dp0condabin\micromamba.bat" deactivate @CALL "%~dp0condabin\micromamba.bat" deactivate
@ -112,7 +114,7 @@ REM Create pixie-env environment
@CALL "%~dp0condabin\micromamba.bat" activate pixie-env @CALL "%~dp0condabin\micromamba.bat" activate pixie-env
@CALL cd %PROJECT_DIR%\ext\PIXIE\ @CALL cd %PROJECT_DIR%\ext\PIXIE\
@CALL python -m pip install --upgrade pip @CALL python -m pip install --upgrade pip
@CALL python -m pip install torch==2.1.1+cu118 torchvision torchaudio pytorch-cuda --index-url https://download.pytorch.org/whl/cu118 --force-reinstall --no-cache-dir @CALL python -m pip install torch==2.1.1+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --force-reinstall --no-cache-dir
@CALL python -m fvcore kornia matplotlib pytorch3d pyyaml @CALL python -m fvcore kornia matplotlib pytorch3d pyyaml
@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%