diff --git a/activate_gaussian_splatting_hair.bat b/activate_gaussian_splatting_hair.bat index 5c32621..896fb48 100644 --- a/activate_gaussian_splatting_hair.bat +++ b/activate_gaussian_splatting_hair.bat @@ -1,15 +1,18 @@ @echo off -SET MICROMAMBA_EXE=%~dp0micromamba.exe -SET MAMBA_ROOT_PREFIX=%~dp0micromamba + +REM 设置环境变量 SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8" -SET PATH=%CUDA_HOME%\bin;%PATH% -SET PROJECT_DIR=%~dp0 -SET PYTHONDONTWRITEBYTECODE=1 -SET TORCH_HOME=%PROJECT_DIR%\cache\torch -SET HF_HOME=%PROJECT_DIR%\cache\huggingface -SET GDOWN_CACHE=%PROJECT_DIR%\cache\gdown +SET PATH="%CUDA_HOME%\bin;%PATH%" SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6" -SET VS2019_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" -SET VS2019_VCVARS="%VS2019_DIR%\VC\Auxiliary\Build\vcvars64.bat" +SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" +SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat" +SET PROJECT_DIR=%~dp0 +SET MICROMAMBA_EXE=%~dp0micromamba.exe +SET MAMBA_ROOT_PREFIX=%PROJECT_DIR%\micromamba +SET PYTHONDONTWRITEBYTECODE=1 +SET GDOWN_CACHE=cache\gdown +SET TORCH_HOME=cache\torch +SET HF_HOME=cache\huggingface CALL "%MICROMAMBA_EXE%" activate -p %MAMBA_ROOT_PREFIX%\envs\gaussian_splatting_hair +IF EXIST %VS_VCVARS% CALL %VS_VCVARS% diff --git a/activate_matte_anything.bat b/activate_matte_anything.bat index b9085b1..980858f 100644 --- a/activate_matte_anything.bat +++ b/activate_matte_anything.bat @@ -1,18 +1,18 @@ @echo off -SET MICROMAMBA_EXE=%~dp0micromamba.exe -SET MAMBA_ROOT_PREFIX=%~dp0micromamba + +REM 设置环境变量 SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8" -SET PATH=%CUDA_HOME%\bin;%PATH% -SET PROJECT_DIR=%~dp0 -SET PYTHONDONTWRITEBYTECODE=1 -SET TORCH_HOME=%PROJECT_DIR%\cache\torch -SET HF_HOME=%PROJECT_DIR%\cache\huggingface -SET GDOWN_CACHE=%PROJECT_DIR%\cache\gdown +SET PATH="%CUDA_HOME%\bin;%PATH%" SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6" -SET VS2019_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" -SET VS2019_VCVARS="%VS2019_DIR%\VC\Auxiliary\Build\vcvars64.bat" -SET DISTUTILS_USE_SDK=1 -SET MSSdk=1 +SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" +SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat" +SET PROJECT_DIR=%~dp0 +SET MICROMAMBA_EXE=%~dp0micromamba.exe +SET MAMBA_ROOT_PREFIX=%PROJECT_DIR%\micromamba +SET PYTHONDONTWRITEBYTECODE=1 +SET GDOWN_CACHE=cache\gdown +SET TORCH_HOME=cache\torch +SET HF_HOME=cache\huggingface CALL "%MICROMAMBA_EXE%" activate -p %MAMBA_ROOT_PREFIX%\envs\matte_anything -IF EXIST %VS2019_VCVARS% CALL %VS2019_VCVARS% +IF EXIST %VS_VCVARS% CALL %VS_VCVARS% diff --git a/activate_openpose.bat b/activate_openpose.bat index e82216a..171e0ea 100644 --- a/activate_openpose.bat +++ b/activate_openpose.bat @@ -1,15 +1,18 @@ @echo off -SET MICROMAMBA_EXE=%~dp0micromamba.exe -SET MAMBA_ROOT_PREFIX=%~dp0micromamba + +REM 设置环境变量 SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8" -SET PATH=%CUDA_HOME%\bin;%PATH% -SET PROJECT_DIR=%~dp0 -SET PYTHONDONTWRITEBYTECODE=1 -SET TORCH_HOME=%PROJECT_DIR%\cache\torch -SET HF_HOME=%PROJECT_DIR%\cache\huggingface -SET GDOWN_CACHE=%PROJECT_DIR%\cache\gdown +SET PATH="%CUDA_HOME%\bin;%PATH%" SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6" -SET VS2019_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" -SET VS2019_VCVARS="%VS2019_DIR%\VC\Auxiliary\Build\vcvars64.bat" +SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" +SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat" +SET PROJECT_DIR=%~dp0 +SET MICROMAMBA_EXE=%~dp0micromamba.exe +SET MAMBA_ROOT_PREFIX=%PROJECT_DIR%\micromamba +SET PYTHONDONTWRITEBYTECODE=1 +SET GDOWN_CACHE=cache\gdown +SET TORCH_HOME=cache\torch +SET HF_HOME=cache\huggingface CALL "%MICROMAMBA_EXE%" activate -p %MAMBA_ROOT_PREFIX%\envs\openpose +IF EXIST %VS_VCVARS% CALL %VS_VCVARS% diff --git a/activate_pixie-env.bat b/activate_pixie-env.bat index 0d79c5f..8b758ab 100644 --- a/activate_pixie-env.bat +++ b/activate_pixie-env.bat @@ -1,15 +1,18 @@ @echo off -SET MICROMAMBA_EXE=%~dp0micromamba.exe -SET MAMBA_ROOT_PREFIX=%~dp0micromamba + +REM 设置环境变量 SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8" -SET PATH=%CUDA_HOME%\bin;%PATH% -SET PROJECT_DIR=%~dp0 -SET PYTHONDONTWRITEBYTECODE=1 -SET TORCH_HOME=%PROJECT_DIR%\cache\torch -SET HF_HOME=%PROJECT_DIR%\cache\huggingface -SET GDOWN_CACHE=%PROJECT_DIR%\cache\gdown +SET PATH="%CUDA_HOME%\bin;%PATH%" SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6" -SET VS2019_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" -SET VS2019_VCVARS="%VS2019_DIR%\VC\Auxiliary\Build\vcvars64.bat" +SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" +SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat" +SET PROJECT_DIR=%~dp0 +SET MICROMAMBA_EXE=%~dp0micromamba.exe +SET MAMBA_ROOT_PREFIX=%PROJECT_DIR%\micromamba +SET PYTHONDONTWRITEBYTECODE=1 +SET GDOWN_CACHE=cache\gdown +SET TORCH_HOME=cache\torch +SET HF_HOME=cache\huggingface CALL "%MICROMAMBA_EXE%" activate -p %MAMBA_ROOT_PREFIX%\envs\pixie-env +IF EXIST %VS_VCVARS% CALL %VS_VCVARS% diff --git a/download_resource.bat b/download_resource.bat index 3cf7503..0b59929 100644 --- a/download_resource.bat +++ b/download_resource.bat @@ -1,6 +1,18 @@ @echo off -set PROJECT_DIR=%~dp0 -cd %PROJECT_DIR% + +REM 设置环境变量 +SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8" +SET PATH="%CUDA_HOME%\bin;%PATH%" +SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6" +SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" +SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat" +SET PROJECT_DIR=%~dp0 +SET MICROMAMBA_EXE=%~dp0micromamba.exe +SET MAMBA_ROOT_PREFIX=%PROJECT_DIR%\micromamba +SET PYTHONDONTWRITEBYTECODE=1 +SET GDOWN_CACHE=cache\gdown +SET TORCH_HOME=cache\torch +SET HF_HOME=cache\huggingface REM 检查是否安装了python if not exist "%PROJECT_DIR%\venv\Scripts\python.exe" ( diff --git a/install.bat b/install.bat index 78a78d3..e1536c5 100644 --- a/install.bat +++ b/install.bat @@ -2,17 +2,18 @@ setlocal enabledelayedexpansion REM 设置环境变量 -SET MICROMAMBA_EXE=%~dp0micromamba.exe -SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8 -SET PATH=%CUDA_HOME%\bin;%PATH% +SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8" +SET PATH="%CUDA_HOME%\bin;%PATH%" +SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6" +SET VS_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" +SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat" SET PROJECT_DIR=%~dp0 +SET MICROMAMBA_EXE=%~dp0micromamba.exe +SET MAMBA_ROOT_PREFIX=%PROJECT_DIR%\micromamba SET PYTHONDONTWRITEBYTECODE=1 SET GDOWN_CACHE=cache\gdown SET TORCH_HOME=cache\torch SET HF_HOME=cache\huggingface -SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6 -SET VS2019_DIR=C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools -SET VS2019_VCVARS=%VS2019_DIR%\VC\Auxiliary\Build\vcvars64.bat REM 检查micromamba IF NOT EXIST "%MICROMAMBA_EXE%" ( @@ -22,7 +23,6 @@ IF NOT EXIST "%MICROMAMBA_EXE%" ( ) REM 设置micromamba根目录 -SET MAMBA_ROOT_PREFIX=%PROJECT_DIR%\micromamba IF NOT EXIST "%MAMBA_ROOT_PREFIX%" mkdir "%MAMBA_ROOT_PREFIX%" REM 初始化micromamba @@ -41,7 +41,7 @@ IF NOT EXIST "%BLENDER_DIR%\" ( ) REM 检查Visual Studio -IF NOT EXIST "%VS2019_VCVARS%" ( +IF NOT EXIST "%VS_VCVARS%" ( echo ERROR: Visual Studio 2019 not found echo Please install Visual Studio 2019 Build Tools with C++ development tools exit /b 1 @@ -97,9 +97,9 @@ IF NOT DEFINED PROJECT_DIR ( exit /b 1 ) -IF NOT DEFINED VS2019_VCVARS ( - echo 错误:未设置VS2019_VCVARS环境变量 - echo 请设置VS2019_VCVARS指向Visual Studio 2019的vcvars64.bat +IF NOT DEFINED VS_VCVARS ( + echo 错误:未设置VS_VCVARS环境变量 + echo 请设置VS_VCVARS指向Visual Studio 2019的vcvars64.bat exit /b 1 ) @@ -253,7 +253,7 @@ CALL activate_openpose.bat cd %PROJECT_DIR%\ext\openpose mkdir build 2>nul cd build -CALL "%VS2019_VCVARS%" +CALL "%VS_VCVARS%" cmake .. -DBUILD_PYTHON=true -DUSE_CUDNN=off -DBUILD_CAFFE=false -G "Visual Studio 16 2019" -A x64 cmake --build . --config Release cd %PROJECT_DIR%