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

Update install.bat

Browse Source
This commit is contained in:
Jeffreytsai1004 2025-02-14 02:18:39 +08:00
parent 60e97925aa
commit 2cad765fbf
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
install.bat
View File

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