diff --git a/install.bat b/install.bat index a4c155c..eb6bb5c 100644 --- a/install.bat +++ b/install.bat @@ -3,15 +3,15 @@ setlocal enabledelayedexpansion REM 设置环境变量 SET MICROMAMBA_EXE=%~dp0micromamba.exe -SET CUDA_HOME="C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8" +SET CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8 SET PATH=%CUDA_HOME%\bin;%PATH% SET PROJECT_DIR=%~dp0 SET PYTHONDONTWRITEBYTECODE=1 SET GDOWN_CACHE=cache\gdown SET TORCH_HOME=cache\torch SET HF_HOME=cache\huggingface -SET BLENDER_DIR="C:\Program Files\Blender Foundation\Blender 3.6" -SET VS2019_DIR="C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools" +SET BLENDER_DIR=C:\Program Files\Blender Foundation\Blender 3.6 +SET VS2019_DIR=C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools SET VS2019_VCVARS=%VS2019_DIR%\VC\Auxiliary\Build\vcvars64.bat REM 检查micromamba @@ -23,17 +23,17 @@ IF NOT EXIST "%MICROMAMBA_EXE%" ( REM 设置micromamba根目录 SET MAMBA_ROOT_PREFIX=%PROJECT_DIR%\micromamba -mkdir %MAMBA_ROOT_PREFIX% +IF NOT EXIST "%MAMBA_ROOT_PREFIX%" mkdir "%MAMBA_ROOT_PREFIX%" REM 初始化micromamba -CALL "%MICROMAMBA_EXE%" shell init --root-prefix=%MAMBA_ROOT_PREFIX% +CALL "%MICROMAMBA_EXE%" shell init --root-prefix "%MAMBA_ROOT_PREFIX%" REM 检查必要的环境和依赖 -IF NOT EXIST "%CUDA_HOME%" ( +IF NOT EXIST "%CUDA_HOME%\" ( echo ERROR: CUDA 11.8 not found at %CUDA_HOME% exit /b 1 ) -IF NOT EXIST "%BLENDER_DIR%" ( +IF NOT EXIST "%BLENDER_DIR%\" ( echo ERROR: Blender 3.6 not found at %BLENDER_DIR% echo Please install Blender 3.6 from https://www.blender.org/download/lts/3-6/ exit /b 1 @@ -47,7 +47,7 @@ IF NOT EXIST "%VS2019_VCVARS%" ( ) REM 检查CUDA版本 -nvcc --version | findstr "release 11.8" >nul +nvcc --version 2>nul | findstr "release 11.8" >nul IF %ERRORLEVEL% NEQ 0 ( echo ERROR: CUDA 11.8 not found or version mismatch exit /b 1