Update install.bat
This commit is contained in:
parent
60e97925aa
commit
2cad765fbf
16
install.bat
16
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
|
||||
|
Loading…
Reference in New Issue
Block a user