Update install.bat
This commit is contained in:
parent
f912fd2bf7
commit
8a9c1de43c
123
install.bat
123
install.bat
@ -1,6 +1,4 @@
|
||||
@echo off
|
||||
|
||||
@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 MICROMAMBA_EXE=%PROJECT_DIR%\micromamba.exe
|
||||
@CALL SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\"
|
||||
@ -9,110 +7,81 @@
|
||||
@CALL SET VS_VCVARS="%VS_DIR%\VC\Auxiliary\Build\vcvars64.bat"
|
||||
@CALL SET PATH=%CUDA_HOME%\bin;%PROJECT_DIR%\condabin;%PATH%
|
||||
|
||||
IF NOT EXIST "%CUDA_HOME%" (
|
||||
echo ERROR: CUDA_HOME path does not exist: %CUDA_HOME%
|
||||
exit /b 1
|
||||
)
|
||||
IF NOT EXIST "%BLENDER_DIR%" (
|
||||
echo ERROR: BLENDER_DIR path does not exist: %BLENDER_DIR%
|
||||
exit /b 1
|
||||
)
|
||||
IF NOT EXIST "%VS_DIR%" (
|
||||
echo ERROR: VS_DIR path does not exist: %VS_DIR%
|
||||
exit /b 1
|
||||
)
|
||||
IF NOT EXIST "%MICROMAMBA_EXE%" (
|
||||
echo ERROR: micromamba not found at %MICROMAMBA_EXE%
|
||||
echo Please install micromamba from https://mamba.readthedocs.io/en/latest/installation.html
|
||||
exit /b 1
|
||||
)
|
||||
|
||||
{{ Gaussian Splatting Hair 环境 }}
|
||||
@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 -c nvidia -r "%~dp0\" -y
|
||||
@CALL condabin\micromamba.bat activate gaussian_splatting_hair
|
||||
@CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair
|
||||
@CALL python -m pip install pip==23.3.1
|
||||
@CALL python -m pip install gdown==5.2.0
|
||||
@CALL python -m pip install -r requirements.txt
|
||||
|
||||
@CALL cd %PROJECT_DIR%\ext
|
||||
@CALL git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose --depth 1
|
||||
@CALL cd %PROJECT_DIR%\ext\openpose
|
||||
@CALL git submodule update --init --recursive --remote
|
||||
@CALL cd %PROJECT_DIR%\ext
|
||||
@CALL git clone https://github.com/hustvl/Matte-Anything
|
||||
@CALL cd %PROJECT_DIR%\ext\Matte-Anything
|
||||
@CALL git clone https://github.com/IDEA-Research/GroundingDINO.git
|
||||
@CALL cd %PROJECT_DIR%\ext
|
||||
@CALL git clone git@github.com:egorzakharov/NeuralHaircut.git --recursive
|
||||
@CALL cd %PROJECT_DIR%\ext
|
||||
@CALL git clone https://github.com/facebookresearch/pytorch3d
|
||||
@CALL cd %PROJECT_DIR%\ext\pytorch3d
|
||||
@CALL git checkout 2f11ddc5ee7d6bd56f2fb6744a16776fab6536f7
|
||||
@CALL cd %PROJECT_DIR%\ext
|
||||
@CALL git clone https://github.com/camenduru/simple-knn
|
||||
@CALL cd %PROJECT_DIR%\ext\diff_gaussian_rasterization_hair\third_party
|
||||
@CALL git clone https://github.com/g-truc/glm
|
||||
@CALL cd %PROJECT_DIR%\ext\diff_gaussian_rasterization_hair\third_party\glm
|
||||
@CALL git checkout 5c46b9c07008ae65cb81ab79cd677ecc1934b903
|
||||
@CALL cd %PROJECT_DIR%\ext
|
||||
@CALL git clone --recursive https://github.com/NVIDIAGameWorks/kaolin
|
||||
@CALL cd %PROJECT_DIR%\ext\kaolin
|
||||
@CALL git checkout v0.15.0
|
||||
@CALL cd %PROJECT_DIR%\ext
|
||||
@CALL git clone https://github.com/SSL92/hyperIQA %PROJECT_DIR%\ext\hyperIQA
|
||||
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose --depth 1
|
||||
@CALL cd %PROJECT_DIR%\ext\openpose && git submodule update --init --recursive --remote
|
||||
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/hustvl/Matte-Anything
|
||||
@CALL cd %PROJECT_DIR%\ext\Matte-Anything && git clone https://github.com/IDEA-Research/GroundingDINO.git
|
||||
@CALL cd %PROJECT_DIR%\ext && git clone git@github.com:egorzakharov/NeuralHaircut.git --recursive
|
||||
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/facebookresearch/pytorch3d
|
||||
@CALL cd %PROJECT_DIR%\ext\pytorch3d && git checkout 2f11ddc5ee7d6bd56f2fb6744a16776fab6536f7
|
||||
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/camenduru/simple-knn
|
||||
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/g-truc/glm
|
||||
@CALL cd %PROJECT_DIR%\ext\diff_gaussian_rasterization_hair\third_party\glm && git checkout 5c46b9c07008ae65cb81ab79cd677ecc1934b903
|
||||
@CALL cd %PROJECT_DIR%\ext && git clone --recursive https://github.com/NVIDIAGameWorks/kaolin
|
||||
@CALL cd %PROJECT_DIR%\ext\kaolin && git checkout v0.15.0
|
||||
@CALL cd %PROJECT_DIR%\ext && git clone https://github.com/SSL92/hyperIQA
|
||||
|
||||
@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\pretrained_models\diffusion_prior %PROJECT_DIR%\ext\NeuralHaircut\pretrained_models\diffusion_prior
|
||||
@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\PIXIE %PROJECT_DIR%\ext\NeuralHaircut\PIXIE
|
||||
@CALL copy %PROJECT_DIR%\resource\NeuralHaircut\ %PROJECT_DIR%\ext\NeuralHaircut\
|
||||
@CALL copy %PROJECT_DIR%\resource\hyperIQA\pretrained %PROJECT_DIR%\ext\hyperIQA\pretrained
|
||||
@CALL copy %PROJECT_DIR%\resource\openpose %PROJECT_DIR%\ext\openpose
|
||||
@CALL cd %PROJECT_DIR%
|
||||
@CALL condabin\micromamba.bat deactivate gaussian_splatting_hair
|
||||
@CALL "%~dp0condabin\micromamba.bat" deactivate gaussian_splatting_hair
|
||||
|
||||
{{ Matte Anything 环境 }}
|
||||
@CALL "%~dp0micromamba.exe" create -n matte_anything python=3.9 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 wget scikit-image gradio=3.46.1 fairscale -c pytorch -c nvidia -c conda-forge -r "%~dp0\" -y
|
||||
@CALL condabin\micromamba.bat activate matte_anything
|
||||
@CALL mkdir %PROJECT_DIR%\ext\Matte-Anything
|
||||
@CALL "%~dp0micromamba.exe" create -n matte_anything ^
|
||||
python=3.9 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 wget scikit-image gradio=3.46.1 fairscale ^
|
||||
-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 git clone git@github.com:facebookresearch/segment-anything.git
|
||||
@CALL cd segment-anything
|
||||
@CALL pip install -e .
|
||||
@CALL git clone git@github.com:facebookresearch/segment-anything.git %PROJECT_DIR%\ext\Matte-Anything\segment-anything
|
||||
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\segment-anything && pip install -e .
|
||||
@CALL cd %PROJECT_DIR%\ext\Matte-Anything
|
||||
@CALL git clone https://github.com/conansherry/detectron2
|
||||
@CALL cd detectron2
|
||||
@CALL pip install -e .
|
||||
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO
|
||||
@CALL pip install -e .
|
||||
@CALL mkdir %PROJECT_DIR%\ext\Matte-Anything\pretrained
|
||||
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\pretrained
|
||||
@CALL copy %PROJECT_DIR%\resource\Matte-Anything\pretrained %PROJECT_DIR%\ext\Matte-Anything\pretrained
|
||||
@CALL git clone https://github.com/conansherry/detectron2 %PROJECT_DIR%\ext\Matte-Anything\detectron2
|
||||
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\detectron2 && pip install -e .
|
||||
@CALL cd %PROJECT_DIR%\ext\Matte-Anything\GroundingDINO && pip install -e .
|
||||
@CALL pip install supervision==0.22.0
|
||||
@CALL cd %PROJECT_DIR%\ext\Matte-Anything && mkdir pretrained
|
||||
@CALL copy %PROJECT_DIR%\resource\Matte-Anything\ %PROJECT_DIR%\ext\Matte-Anything\
|
||||
@CALL cd %PROJECT_DIR%
|
||||
@CALL condabin\micromamba.bat deactivate matte_anything
|
||||
@CALL "%~dp0condabin\micromamba.bat" deactivate matte_anything
|
||||
|
||||
{{ OpenPose 环境 }}
|
||||
@CALL "%~dp0micromamba.exe" create -n openpose cmake=3.20 -c conda-forge -r "%~dp0\" -y
|
||||
@CALL condabin\micromamba.bat activate openpose
|
||||
@CALL cd %PROJECT_DIR%\ext\openpose
|
||||
@CALL git submodule update --init --recursive --remote
|
||||
@CALL "%~dp0condabin\micromamba.bat" deactivate matte_anything
|
||||
@CALL "%~dp0condabin\micromamba.bat" activate gaussian_splatting_hair
|
||||
@CALL copy %PROJECT_DIR%\resource\openpose %PROJECT_DIR%\ext\openpose
|
||||
@CALL "%~dp0condabin\micromamba.bat" deactivate gaussian_splatting_hair
|
||||
@CALL git submodule update --init --recursive --remote
|
||||
@CALL "%~dp0condabin\micromamba.bat" deactivate openpose
|
||||
@CALL "%~dp0micromamba.exe" create -n openpose cmake=3.20 -c conda-forge -r "%~dp0\" -y
|
||||
@CALL "%~dp0condabin\micromamba.bat" activate openpose
|
||||
@CALL cd %PROJECT_DIR%\ext\openpose
|
||||
@CALL mkdir build
|
||||
@CALL cd build
|
||||
@CALL %VS_VCVARS%
|
||||
@CALL cmake .. -G "Visual Studio 17 2022" -A x64 -T host=x64 -DBUILD_PYTHON=true -DUSE_CUDNN=off
|
||||
@CALL cmake --build . --config Release
|
||||
@CALL cd %PROJECT_DIR%
|
||||
@CALL condabin\micromamba.bat deactivate openpose
|
||||
@CALL "%~dp0condabin\micromamba.bat" deactivate openpose
|
||||
|
||||
{{ PIXIE 环境 }}
|
||||
@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.8 pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 fvcore pytorch3d==0.7.5 kornia matplotlib -c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y
|
||||
@CALL condabin\micromamba.bat activate pixie-env
|
||||
@CALL cd %PROJECT_DIR%\ext
|
||||
@CALL git clone https://github.com/yfeng95/PIXIE
|
||||
@CALL "%~dp0micromamba.exe" create -n pixie-env python=3.8 pytorch==2.0.0 torchvision torchaudio ^
|
||||
pytorch-cuda=11.8 fvcore pytorch3d==0.7.5 kornia matplotlib ^
|
||||
-c pytorch -c nvidia -c fvcore -c conda-forge -c pytorch3d -r "%~dp0\" -y
|
||||
@CALL "%~dp0condabin\micromamba.bat" activate pixie-env
|
||||
@CALL cd %PROJECT_DIR%\ext\PIXIE
|
||||
@CALL chmod +x fetch_model.sh && ./fetch_model.sh
|
||||
@CALL copy %PROJECT_DIR%\resource\PIXIE\pixie_data %PROJECT_DIR%\ext\PIXIE\data
|
||||
@CALL pip install pyyaml==5.4.1
|
||||
@CALL pip install git+https://github.com/1adrianb/face-alignment.git@54623537fd9618ca7c15688fd85aba706ad92b59
|
||||
@CALL cd %PROJECT_DIR%
|
||||
@CALL condabin\micromamba.bat deactivate pixie-env
|
||||
@CALL "%~dp0condabin\micromamba.bat" deactivate pixie-env
|
||||
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user