This commit is contained in:
Jeffreytsai1004 2025-02-23 00:37:03 +08:00
parent 4b4c776842
commit 32c4d60727

View File

@ -2,10 +2,10 @@
setlocal enabledelayedexpansion setlocal enabledelayedexpansion
REM Set environment variables REM Set environment variables
set "PYTHONDONTWRITEBYTECODE=1" @CALL set "PYTHONDONTWRITEBYTECODE=1"
set "GDOWN_CACHE=%PROJECT_DIR%\cache\gdown" @CALL set "GDOWN_CACHE=%PROJECT_DIR%\cache\gdown"
set "TORCH_HOME=%PROJECT_DIR%\cache\torch" @CALL set "TORCH_HOME=%PROJECT_DIR%\cache\torch"
set "HF_HOME=%PROJECT_DIR%\cache\huggingface" @CALL set "HF_HOME=%PROJECT_DIR%\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"
@ -17,8 +17,7 @@ set "HF_HOME=%PROJECT_DIR%\cache\huggingface"
REM Create gaussian_splatting_hair environment REM Create gaussian_splatting_hair environment
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%
@CALL "%~dp0micromamba.exe" create -n gaussian_splatting_hair python=3.9 ^ @CALL "%~dp0micromamba.exe" create -n gaussian_splatting_hair python=3.9 pytorch-cuda=11.8 ^
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 ^ 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 ^ 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 ^ fvcore=0.1.5 toml=0.10.2 tqdm=4.66.5 gdown=5.2.0 colmap=3.10 ^
@ -27,6 +26,7 @@ REM Create gaussian_splatting_hair environment
@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 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 @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
@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
REM Clone repositories REM Clone repositories
@CALL cd %PROJECT_DIR% @CALL cd %PROJECT_DIR%