diff --git a/install.bat b/install.bat
index 07715dc..035fb23 100644
--- a/install.bat
+++ b/install.bat
@@ -2,10 +2,10 @@
 setlocal enabledelayedexpansion
 
 REM Set environment variables
-set "PYTHONDONTWRITEBYTECODE=1"
-set "GDOWN_CACHE=%PROJECT_DIR%\cache\gdown"
-set "TORCH_HOME=%PROJECT_DIR%\cache\torch"
-set "HF_HOME=%PROJECT_DIR%\cache\huggingface"
+@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 "%~dp0micromamba.exe" shell init --shell cmd.exe --prefix "%~dp0\"
 @CALL SET PROJECT_DIR="%~dp0"
@@ -17,8 +17,7 @@ set "HF_HOME=%PROJECT_DIR%\cache\huggingface"
 
 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 ^
+@CALL "%~dp0micromamba.exe" create -n gaussian_splatting_hair python=3.9 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 ^
@@ -27,6 +26,7 @@ REM Create gaussian_splatting_hair environment
 @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
+@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
 @CALL cd %PROJECT_DIR%