CGNICO/GaussianHaircut
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Jeffreytsai1004 2025-02-25 08:48:36 +08:00
parent 69a81ec4c3
commit 49d368faed
5 changed files with 253 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
activate_gaussian_splatting_hair.bat
View File

@ -1,2 +1,31 @@
@CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\"
start cmd /k "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair
@CALL SET PROJECT_DIR=%~dp0
@CALL SET EXT_DIR=%PROJECT_DIR%ext
@CALL SET ENV_DIR=%PROJECT_DIR%envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%resource
@CALL SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
@CALL SET CUDA_DIR=%CUDA_HOME%
@CALL SET CUDA_PATH=%CUDA_HOME%
@CALL SET CUDA_PATH_V11_8=%CUDA_HOME%
@CALL SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6
@CALL SET VS_DIR=D:\Program Files\Microsoft Visual Studio\2022\Community
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@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
Echo CUDA_HOME: %CUDA_HOME%
Echo CUDA_DIR: %CUDA_DIR%
Echo PROJECT_DIR: %PROJECT_DIR%
Echo EXT_DIR: %EXT_DIR%
Echo ENV_DIR: %ENV_DIR%
Echo RESOURCE_DIR: %RESOURCE_DIR%
Echo BLENDER_DIR: %BLENDER_DIR%
Echo VS_DIR: %VS_DIR%
Echo VS_VCVARS: %VS_VCVARS%
Echo GDOWN_CACHE: %GDOWN_CACHE%
Echo TORCH_HOME: %TORCH_HOME%
Echo HF_HOME: %HF_HOME%
@CALL SET PATH=%CUDA_HOME%\bin;%ENV_DIR%\gaussian_splatting_hair\bin;%PROJECT_DIR%condabin;%PATH%

29
activate_matte_anything.bat
View File

@ -1,2 +1,31 @@
@CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\"
start cmd /k "%~dp0condabin\micromamba.bat" activate matte_anything
@CALL SET PROJECT_DIR=%~dp0
@CALL SET EXT_DIR=%PROJECT_DIR%ext
@CALL SET ENV_DIR=%PROJECT_DIR%envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%resource
@CALL SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
@CALL SET CUDA_DIR=%CUDA_HOME%
@CALL SET CUDA_PATH=%CUDA_HOME%
@CALL SET CUDA_PATH_V11_8=%CUDA_HOME%
@CALL SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6
@CALL SET VS_DIR=D:\Program Files\Microsoft Visual Studio\2022\Community
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@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
Echo CUDA_HOME: %CUDA_HOME%
Echo CUDA_DIR: %CUDA_DIR%
Echo PROJECT_DIR: %PROJECT_DIR%
Echo EXT_DIR: %EXT_DIR%
Echo ENV_DIR: %ENV_DIR%
Echo RESOURCE_DIR: %RESOURCE_DIR%
Echo BLENDER_DIR: %BLENDER_DIR%
Echo VS_DIR: %VS_DIR%
Echo VS_VCVARS: %VS_VCVARS%
Echo GDOWN_CACHE: %GDOWN_CACHE%
Echo TORCH_HOME: %TORCH_HOME%
Echo HF_HOME: %HF_HOME%
@CALL SET PATH=%CUDA_HOME%\bin;%ENV_DIR%\matte_anything\bin;%PROJECT_DIR%condabin;%PATH%

29
activate_openpose.bat
View File

@ -1,2 +1,31 @@
@CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\"
start cmd /k "%~dp0condabin\micromamba.bat" activate openpose
@CALL SET PROJECT_DIR=%~dp0
@CALL SET EXT_DIR=%PROJECT_DIR%ext
@CALL SET ENV_DIR=%PROJECT_DIR%envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%resource
@CALL SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
@CALL SET CUDA_DIR=%CUDA_HOME%
@CALL SET CUDA_PATH=%CUDA_HOME%
@CALL SET CUDA_PATH_V11_8=%CUDA_HOME%
@CALL SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6
@CALL SET VS_DIR=D:\Program Files\Microsoft Visual Studio\2022\Community
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@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
Echo CUDA_HOME: %CUDA_HOME%
Echo CUDA_DIR: %CUDA_DIR%
Echo PROJECT_DIR: %PROJECT_DIR%
Echo EXT_DIR: %EXT_DIR%
Echo ENV_DIR: %ENV_DIR%
Echo RESOURCE_DIR: %RESOURCE_DIR%
Echo BLENDER_DIR: %BLENDER_DIR%
Echo VS_DIR: %VS_DIR%
Echo VS_VCVARS: %VS_VCVARS%
Echo GDOWN_CACHE: %GDOWN_CACHE%
Echo TORCH_HOME: %TORCH_HOME%
Echo HF_HOME: %HF_HOME%
@CALL SET PATH=%CUDA_HOME%\bin;%ENV_DIR%\openpose\bin;%PROJECT_DIR%condabin;%PATH%

29
activate_pixie-env.bat
View File

@ -1,2 +1,31 @@
@CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\"
start cmd /k "%~dp0condabin\micromamba.bat" activate pixie-env
@CALL SET PROJECT_DIR=%~dp0
@CALL SET EXT_DIR=%PROJECT_DIR%ext
@CALL SET ENV_DIR=%PROJECT_DIR%envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%resource
@CALL SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
@CALL SET CUDA_DIR=%CUDA_HOME%
@CALL SET CUDA_PATH=%CUDA_HOME%
@CALL SET CUDA_PATH_V11_8=%CUDA_HOME%
@CALL SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6
@CALL SET VS_DIR=D:\Program Files\Microsoft Visual Studio\2022\Community
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@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
Echo CUDA_HOME: %CUDA_HOME%
Echo CUDA_DIR: %CUDA_DIR%
Echo PROJECT_DIR: %PROJECT_DIR%
Echo EXT_DIR: %EXT_DIR%
Echo ENV_DIR: %ENV_DIR%
Echo RESOURCE_DIR: %RESOURCE_DIR%
Echo BLENDER_DIR: %BLENDER_DIR%
Echo VS_DIR: %VS_DIR%
Echo VS_VCVARS: %VS_VCVARS%
Echo GDOWN_CACHE: %GDOWN_CACHE%
Echo TORCH_HOME: %TORCH_HOME%
Echo HF_HOME: %HF_HOME%
@CALL SET PATH=%CUDA_HOME%\bin;%ENV_DIR%\pixie-env\bin;%PROJECT_DIR%condabin;%PATH%

162
install.bat
View File

@ -3,21 +3,22 @@
@CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\"
Echo ============= SETTING UP GLOBAL VARIABLES =============
@CALL SET PROJECT_DIR=D:\Dev\AI\Gitea\GaussianHaircut
@CALL SET EXT_DIR=%PROJECT_DIR%\ext
@CALL SET ENV_DIR=%PROJECT_DIR%\envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%\resource
@CALL SET PROJECT_DIR=%~dp0
@CALL SET EXT_DIR=%PROJECT_DIR%ext
@CALL SET ENV_DIR=%PROJECT_DIR%envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%resource
@CALL SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
@CALL SET CUDA_DIR=%CUDA_HOME%
@CALL SET CUDA_PATH=%CUDA_HOME%
@CALL SET CUDA_PATH_V11_8=%CUDA_HOME%
@CALL SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6
@CALL SET VS_DIR=D:\Program Files\Microsoft Visual Studio\2022\Community
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@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 PATH=%CUDA_HOME%\bin;%PROJECT_DIR%condabin;%PATH%
@CALL set GDOWN_CACHE=%PROJECT_DIR%cache\gdown
@CALL set TORCH_HOME=%PROJECT_DIR%cache\torch
@CALL set HF_HOME=%PROJECT_DIR%cache\huggingface
Echo CUDA_HOME: %CUDA_HOME%
Echo CUDA_DIR: %CUDA_DIR%
Echo PROJECT_DIR: %PROJECT_DIR%
@ -30,28 +31,46 @@ Echo VS_VCVARS: %VS_VCVARS%
Echo GDOWN_CACHE: %GDOWN_CACHE%
Echo TORCH_HOME: %TORCH_HOME%
Echo HF_HOME: %HF_HOME%
REM 验证 CUDA 设置
@CALL nvcc --version
REM 设置 Visual Studio 环境
@CALL SET VS_DIR=C:\Program Files\Microsoft Visual Studio\2022\Professional
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL IF EXIST "%VS_VCVARS%" CALL "%VS_VCVARS%"
REM 验证编译器
@CALL WHERE cl.exe
@CALL SET PATH=%CUDA_HOME%\bin;%PROJECT_DIR%condabin;%PATH%
Echo ============= CREATING GAUSSIAN_SPLATTING_HAIR ENVIRONMENT =============
@CALL cd %PROJECT_DIR%
@CALL "%~dp0micromamba.exe" create -n gaussian_splatting_hair python=3.9 ^
-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 SET PATH=%ENV_DIR%\gaussian_splatting_hair\bin;%PATH%
@CALL SET PROJECT_DIR=%~dp0
@CALL SET EXT_DIR=%PROJECT_DIR%ext
@CALL SET ENV_DIR=%PROJECT_DIR%envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%resource
@CALL SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
@CALL SET CUDA_DIR=%CUDA_HOME%
@CALL SET CUDA_PATH=%CUDA_HOME%
@CALL SET CUDA_PATH_V11_8=%CUDA_HOME%
@CALL SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6
@CALL SET VS_DIR=D:\Program Files\Microsoft Visual Studio\2022\Community
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@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
Echo CUDA_HOME: %CUDA_HOME%
Echo CUDA_DIR: %CUDA_DIR%
Echo PROJECT_DIR: %PROJECT_DIR%
Echo EXT_DIR: %EXT_DIR%
Echo ENV_DIR: %ENV_DIR%
Echo RESOURCE_DIR: %RESOURCE_DIR%
Echo BLENDER_DIR: %BLENDER_DIR%
Echo VS_DIR: %VS_DIR%
Echo VS_VCVARS: %VS_VCVARS%
Echo GDOWN_CACHE: %GDOWN_CACHE%
Echo TORCH_HOME: %TORCH_HOME%
Echo HF_HOME: %HF_HOME%
@CALL SET PATH=%CUDA_HOME%\bin;%ENV_DIR%\gaussian_splatting_hair\bin;%PROJECT_DIR%condabin;%PATH%
@CALL cd %PROJECT_DIR%
@CALL python -m pip install --upgrade pip
@CALL python -m pip install torch==2.6.0+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --force-reinstall --no-cache-dir
@CALL python -m pip install cmake pyhocon icecream einops accelerate jsonmerge easydict iopath tensorboardx scikit-image gdown face-alignment clip torchdiffeq torchsde resize-right clean-fid
@CALL python -m pip install cmake pyhocon icecream einops accelerate jsonmerge easydict iopath tensorboardx scikit-image gdown face-alignment clip torchdiffeq torchsde resize-right clean-fid pyside2 future
@CALL cd %PROJECT_DIR%
@CALL mkdir %EXT_DIR%
@ -63,7 +82,7 @@ Echo ============= CREATING GAUSSIAN_SPLATTING_HAIR ENVIRONMENT =============
@CALL git clone https://github.com/g-truc/glm %EXT_DIR%\glm -b 5c46b9c07008ae65cb81ab79cd677ecc1934b903
@CALL git clone --recursive https://github.com/NVIDIAGameWorks/kaolin %EXT_DIR%\kaolin -b v0.15.0
@CALL git clone https://github.com/facebookresearch/segment-anything.git %EXT_DIR%\Matte-Anything\segment-anything
@CALL git clone git+https://github.com/facebookresearch/detectron2.git %EXT_DIR%\Matte-Anything\detectron2
@CALL git clone https://github.com/facebookresearch/detectron2.git %EXT_DIR%\Matte-Anything\detectron2
@CALL git clone https://github.com/SSL92/hyperIQA %EXT_DIR%\hyperIQA
@CALL git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose %EXT_DIR%\openpose --depth 1
@CALL git clone https://github.com/yfeng95/PIXIE %EXT_DIR%\PIXIE
@ -92,27 +111,84 @@ Echo ============= CREATING MATTE_ANYTHING ENVIRONMENT =============
@CALL "%~dp0micromamba.exe" create -n matte_anything python=3.9 ^
-c pytorch -c nvidia -c conda-forge -r "%~dp0\" -y
@CALL "%~dp0condabin\micromamba.bat" activate matte_anything
@CALL SET PATH=%ENV_DIR%\matte_anything\bin;%PATH%
@CALL SET PROJECT_DIR=%~dp0
@CALL SET EXT_DIR=%PROJECT_DIR%ext
@CALL SET ENV_DIR=%PROJECT_DIR%envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%resource
@CALL SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
@CALL SET CUDA_DIR=%CUDA_HOME%
@CALL SET CUDA_PATH=%CUDA_HOME%
@CALL SET CUDA_PATH_V11_8=%CUDA_HOME%
@CALL SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6
@CALL SET VS_DIR=D:\Program Files\Microsoft Visual Studio\2022\Community
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@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
Echo CUDA_HOME: %CUDA_HOME%
Echo CUDA_DIR: %CUDA_DIR%
Echo PROJECT_DIR: %PROJECT_DIR%
Echo EXT_DIR: %EXT_DIR%
Echo ENV_DIR: %ENV_DIR%
Echo RESOURCE_DIR: %RESOURCE_DIR%
Echo BLENDER_DIR: %BLENDER_DIR%
Echo VS_DIR: %VS_DIR%
Echo VS_VCVARS: %VS_VCVARS%
Echo GDOWN_CACHE: %GDOWN_CACHE%
Echo TORCH_HOME: %TORCH_HOME%
Echo HF_HOME: %HF_HOME%
@CALL SET PATH=%CUDA_HOME%\bin;%ENV_DIR%\matte_anything\bin;%PROJECT_DIR%condabin;%PATH%
@CALL cd %EXT_DIR%\Matte-Anything\
@CALL python -m pip install --upgrade pip
@CALL python -m pip install torch==2.6.0+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --force-reinstall --no-cache-dir
@CALL python -m pip install cmake mkl setuptools easydict wget scikit-image gradio fairscale pyside2 future pillow
@CALL python -m pip install ninja
@CALL cd %EXT_DIR%\Matte-Anything\segment-anything
@CALL pip install -e .
@CALL python -m pip install -e . --global-option="build_ext" --global-option="-j4"
@CALL cd %EXT_DIR%\Matte-Anything\detectron2
@CALL pip install -e .
@CALL python -m pip install -e . --global-option="build_ext" --global-option="-j4"
@CALL cd %EXT_DIR%\Matte-Anything\GroundingDINO
@CALL pip install -e .
@CALL python -m pip install -e . --global-option="build_ext" --global-option="-j4"
@CALL cd %EXT_DIR%\Matte-Anything
@CALL pip install supervision==0.22.0
@CALL python -m pip install supervision==0.22.0
@CALL cd %PROJECT_DIR%
Echo ============= CREATING OPENPOSE ENVIRONMENT =============
@CALL cd %PROJECT_DIR%
@CALL "%~dp0micromamba.exe" create -n openpose python=3.9 -c conda-forge -r "%~dp0\" -y
@CALL "%~dp0condabin\micromamba.bat" activate openpose
@CALL SET PROJECT_DIR=%~dp0
@CALL SET EXT_DIR=%PROJECT_DIR%ext
@CALL SET ENV_DIR=%PROJECT_DIR%envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%resource
@CALL SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
@CALL SET CUDA_DIR=%CUDA_HOME%
@CALL SET CUDA_PATH=%CUDA_HOME%
@CALL SET CUDA_PATH_V11_8=%CUDA_HOME%
@CALL SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6
@CALL SET VS_DIR=D:\Program Files\Microsoft Visual Studio\2022\Community
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@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
Echo CUDA_HOME: %CUDA_HOME%
Echo CUDA_DIR: %CUDA_DIR%
Echo PROJECT_DIR: %PROJECT_DIR%
Echo EXT_DIR: %EXT_DIR%
Echo ENV_DIR: %ENV_DIR%
Echo RESOURCE_DIR: %RESOURCE_DIR%
Echo BLENDER_DIR: %BLENDER_DIR%
Echo VS_DIR: %VS_DIR%
Echo VS_VCVARS: %VS_VCVARS%
Echo GDOWN_CACHE: %GDOWN_CACHE%
Echo TORCH_HOME: %TORCH_HOME%
Echo HF_HOME: %HF_HOME%
@CALL SET PATH=%CUDA_HOME%\bin;%ENV_DIR%\openpose\bin;%PROJECT_DIR%condabin;%PATH%
@CALL python -m pip install ninja cmake
@CALL SET PATH=%ENV_DIR%\openpose\bin;%PATH%
@CALL cd %EXT_DIR%\openpose
@CALL mkdir build
@CALL cd build
@ -133,7 +209,35 @@ Echo ============= CREATING PIXIE-ENV ENVIRONMENT =============
@CALL cd %PROJECT_DIR%
@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.9 -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y
@CALL "%~dp0condabin\micromamba.bat" activate pixie-env
@CALL SET PATH=%ENV_DIR%\pixie-env\bin;%PATH%
@CALL SET PROJECT_DIR=%~dp0
@CALL SET EXT_DIR=%PROJECT_DIR%ext
@CALL SET ENV_DIR=%PROJECT_DIR%envs
@CALL SET RESOURCE_DIR=%PROJECT_DIR%resource
@CALL SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
@CALL SET CUDA_DIR=%CUDA_HOME%
@CALL SET CUDA_PATH=%CUDA_HOME%
@CALL SET CUDA_PATH_V11_8=%CUDA_HOME%
@CALL SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6
@CALL SET VS_DIR=D:\Program Files\Microsoft Visual Studio\2022\Community
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@CALL SET VS_VCVARS=%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat
@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
Echo CUDA_HOME: %CUDA_HOME%
Echo CUDA_DIR: %CUDA_DIR%
Echo PROJECT_DIR: %PROJECT_DIR%
Echo EXT_DIR: %EXT_DIR%
Echo ENV_DIR: %ENV_DIR%
Echo RESOURCE_DIR: %RESOURCE_DIR%
Echo BLENDER_DIR: %BLENDER_DIR%
Echo VS_DIR: %VS_DIR%
Echo VS_VCVARS: %VS_VCVARS%
Echo GDOWN_CACHE: %GDOWN_CACHE%
Echo TORCH_HOME: %TORCH_HOME%
Echo HF_HOME: %HF_HOME%
@CALL SET PATH=%CUDA_HOME%\bin;%ENV_DIR%\pixie-env\bin;%PROJECT_DIR%condabin;%PATH%
@CALL cd %EXT_DIR%\PIXIE
@CALL python -m pip install --upgrade pip
@CALL pip install torch==2.6.0+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --force-reinstall --no-cache-dir