diff --git a/activate_gaussian_splatting_hair.bat b/activate_gaussian_splatting_hair.bat index 6fdb47e..24d365d 100644 --- a/activate_gaussian_splatting_hair.bat +++ b/activate_gaussian_splatting_hair.bat @@ -1,2 +1,31 @@ @CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\" -start cmd /k "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair \ No newline at end of file +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% \ No newline at end of file diff --git a/activate_matte_anything.bat b/activate_matte_anything.bat index 1476b69..7b4d1c2 100644 --- a/activate_matte_anything.bat +++ b/activate_matte_anything.bat @@ -1,2 +1,31 @@ @CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\" -start cmd /k "%~dp0condabin\micromamba.bat" activate matte_anything \ No newline at end of file +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% \ No newline at end of file diff --git a/activate_openpose.bat b/activate_openpose.bat index 4df15d0..9dc0ef8 100644 --- a/activate_openpose.bat +++ b/activate_openpose.bat @@ -1,2 +1,31 @@ @CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\" -start cmd /k "%~dp0condabin\micromamba.bat" activate openpose \ No newline at end of file +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% \ No newline at end of file diff --git a/activate_pixie-env.bat b/activate_pixie-env.bat index 147073a..6f2c0a4 100644 --- a/activate_pixie-env.bat +++ b/activate_pixie-env.bat @@ -1,2 +1,31 @@ @CALL "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\" -start cmd /k "%~dp0condabin\micromamba.bat" activate pixie-env \ No newline at end of file +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% \ No newline at end of file diff --git a/install.bat b/install.bat index 8ab1b36..5aa5900 100644 --- a/install.bat +++ b/install.bat @@ -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