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