Update

.gitignore

@@ -24,6 +24,7 @@ Scripts
 pkgs
 envs
 condabin
+cache
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

install.bat

@@ -16,27 +16,50 @@ set "HF_HOME=%PROJECT_DIR%\cache\huggingface"
 @CALL SET PATH=%CUDA_HOME%\bin;%PROJECT_DIR%\condabin;%PATH%
 
 REM Check prerequisites
-where git >nul 2>&1
+where python >nul 2>&1
 if %ERRORLEVEL% neq 0 (
-    echo ERROR: Git not found
+    echo 错误: 未找到Python
     exit /b 1
 )
 
-where cmake >nul 2>&1
+where nvcc >nul 2>&1
 if %ERRORLEVEL% neq 0 (
-    echo ERROR: CMake not found
+    echo 错误: 未找到CUDA工具包
+    exit /b 1
+)
+
+nvcc --version | findstr "release 11.8" >nul 2>&1
+if %ERRORLEVEL% neq 0 (
+    echo 错误: 需要CUDA 11.8版本
+    exit /b 1
+)
+
+if not exist "%BLENDER_DIR%" (
+    echo 错误: 未找到Blender 3.6
+    exit /b 1
+)
+
+if not exist "%VS_DIR%" (
+    echo 错误: 未找到Visual Studio
+    exit /b 1
+)
+
+where git >nul 2>&1
+if %ERRORLEVEL% neq 0 (
+    echo 错误: 未找到 Git
     exit /b 1
 )
 
 REM Create gaussian_splatting_hair environment
+@CALL cd %PROJECT_DIR%
 @CALL "%~dp0micromamba.exe" create -n gaussian_splatting_hair python=3.9 ^
     pytorch=2.1.1 torchvision=0.16.1 torchaudio=2.1.1 pytorch-cuda=11.8 ^
     cmake=3.28.0 pyhocon=0.3.60 icecream=2.1.3 einops=0.6.0 accelerate=0.18.0 ^
     jsonmerge=1.9.0 easydict=1.9 iopath=0.1.10 tensorboardx=2.6 scikit-image=0.20.0 ^
     fvcore=0.1.5 toml=0.10.2 tqdm=4.66.5 gdown=5.2.0 colmap=3.10 ^
     -c pytorch -c nvidia -c conda-forge -c anaconda -c fvcore -c iopath -c bottler -r "%~dp0\" -y
-
 @CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair
+@CALL cd %PROJECT_DIR%
 @CALL python -m pip install --upgrade pip
 @CALL python -m pip install pysdf==0.1.9 clean-fid==0.1.35 face-alignment==1.4.1 clip==0.2.0 torchdiffeq==0.2.3 torchsde==0.2.5 resize-right==0.0.2
 
@@ -81,7 +104,6 @@ REM Create matte_anything environment
     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 scikit-image gradio=3.46.1 ^
     -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 python -m pip install git+https://github.com/facebookresearch/segment-anything.git