-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrun.bat
More file actions
75 lines (63 loc) · 1.64 KB
/
Copy pathrun.bat
File metadata and controls
75 lines (63 loc) · 1.64 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
@echo off
REM LLM-Local 启动脚本 (Windows)
setlocal enabledelayedexpansion
echo.
echo 🤖 LLM-Local 启动器
echo ====================
echo.
REM 检查 Python
python --version >nul 2>&1
if errorlevel 1 (
echo ❌ 未找到 Python。请先安装 Python 3.9 或更高版本。
exit /b 1
)
echo ✅ 找到 Python
for /f "tokens=2" %%i in ('python --version 2^>^&1') do set PYTHON_VERSION=%%i
echo 版本: %PYTHON_VERSION%
REM 检查虚拟环境
if not exist "venv" (
echo ⚠️ 未找到虚拟环境
echo ℹ️ 正在创建虚拟环境...
python -m venv venv
echo ✅ 虚拟环境创建成功
) else (
echo ✅ 找到虚拟环境
)
REM 激活虚拟环境
call venv\Scripts\activate.bat
echo ✅ 虚拟环境已激活
REM 检测平台
echo ℹ️ 检测硬件平台...
nvidia-smi >nul 2>&1
if errorlevel 1 (
set PLATFORM=cpu
echo ℹ️ 未检测到 NVIDIA GPU,将使用 CPU
) else (
set PLATFORM=cuda
echo ℹ️ 检测到 NVIDIA GPU
)
REM 检查依赖
echo ℹ️ 检查依赖...
python -c "import torch" >nul 2>&1
if errorlevel 1 (
echo ⚠️ 未安装推理引擎
echo ℹ️ 正在安装依赖...
if "%PLATFORM%"=="cuda" (
echo ℹ️ 安装 CUDA 依赖...
pip install torch --index-url https://download.pytorch.org/whl/cu118
pip install -r requirements-cuda.txt
) else (
echo ℹ️ 安装基础依赖...
pip install -r requirements.txt
)
echo ✅ 依赖安装完成
) else (
echo ✅ 依赖已安装
)
echo.
echo ✅ 环境准备完成!
echo.
echo ℹ️ 正在启动应用...
echo.
REM 启动应用
python main.py %*