Update install.bat

This commit is contained in:
Jeffreytsai1004 2025-02-23 22:11:30 +08:00
parent 2e28625267
commit 6eb8253a00

View File

@ -3,12 +3,12 @@ setlocal enabledelayedexpansion
REM Set environment variables REM Set environment variables
@CALL set "PYTHONDONTWRITEBYTECODE=1" @CALL set "PYTHONDONTWRITEBYTECODE=1"
@CALL set "GDOWN_CACHE=%PROJECT_DIR%\cache\gdown" @CALL set "GDOWN_CACHE=%~dp0\cache\gdown"
@CALL set "TORCH_HOME=%PROJECT_DIR%\cache\torch" @CALL set "TORCH_HOME=%~dp0\cache\torch"
@CALL set "HF_HOME=%PROJECT_DIR%\cache\huggingface" @CALL set "HF_HOME=%~dp0\cache\huggingface"
@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 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="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" @CALL SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools"
@ -89,7 +89,7 @@ REM Create matte_anything environment
@CALL python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW @CALL python -m gdown 1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW
REM Create openpose environment 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 git submodule update --init --recursive --remote @CALL git submodule update --init --recursive --remote
@ -98,27 +98,27 @@ REM Create openpose environment
@CALL "%~dp0condabin\micromamba.bat" activate openpose @CALL "%~dp0condabin\micromamba.bat" activate openpose
REM 完全清理构建目录 REM 完全清理构建目录
@CALL cd %PROJECT_DIR%\ext\openpose @CALL cd %PROJECT_DIR%ext\openpose
@CALL rmdir /s /q build 2>nul @CALL rmdir /s /q build 2>nul
@CALL del /f /q CMakeCache.txt 2>nul @CALL del /f /q CMakeCache.txt 2>nul
@CALL rmdir /s /q CMakeFiles 2>nul @CALL rmdir /s /q CMakeFiles 2>nul
@CALL mkdir build @CALL mkdir build
@CALL cd build
@CALL set VS_VCVARS_FIXED="%VS_VCVARS:"=%"
@CALL %VS_VCVARS_FIXED%
REM 设置环境变量 REM 设置环境变量
@CALL set "OPENPOSE_ROOT=%PROJECT_DIR%\ext\openpose" @CALL set "OPENPOSE_ROOT=%PROJECT_DIR%ext\openpose"
@CALL set "PYTHON_ROOT=%~dp0micromamba\envs\openpose" @CALL set "PYTHON_ROOT=%PROJECT_DIR%micromamba\envs\openpose"
@CALL set "PYTHON_EXECUTABLE=%PYTHON_ROOT%\python.exe" @CALL set "PYTHON_EXECUTABLE=%PYTHON_ROOT%\python.exe"
@CALL set "PYTHON_LIBRARY=%PYTHON_ROOT%\libs\python39.lib" @CALL set "PYTHON_LIBRARY=%PYTHON_ROOT%\libs\python39.lib"
@CALL set "PYTHON_INCLUDE_DIR=%PYTHON_ROOT%\include" @CALL set "PYTHON_INCLUDE_DIR=%PYTHON_ROOT%\include"
@CALL set "OpenCV_DIR=%PYTHON_ROOT%\Library\cmake\opencv4" @CALL set "OpenCV_DIR=%PYTHON_ROOT%\Library\cmake\opencv4"
REM 确保在正确的目录执行 CMake REM 设置 VS 环境
@CALL cd %PROJECT_DIR%\ext\openpose\build @CALL set VS_VCVARS_FIXED="%VS_VCVARS:"=%"
@CALL cmake %PROJECT_DIR%\ext\openpose -G "Visual Studio 16 2019" -A x64 -T host=x64 ^ @CALL %VS_VCVARS_FIXED%
REM 执行 CMake 构建
@CALL cd %OPENPOSE_ROOT%\build
@CALL cmake %OPENPOSE_ROOT% -G "Visual Studio 16 2019" -A x64 -T host=x64 ^
-DBUILD_PYTHON=true ^ -DBUILD_PYTHON=true ^
-DUSE_CUDNN=off ^ -DUSE_CUDNN=off ^
-DOpenCV_DIR="%OpenCV_DIR%" ^ -DOpenCV_DIR="%OpenCV_DIR%" ^