FLASK和GPU依赖安装

1. 安装所有依赖包

在终端运行以下命令（如果使用 GPU，请确保已安装 CUDA 驱动）：

# 安装 Flask 和其他依赖
pip install flask sentence-transformers torch

# 如果是 GPU 环境，推荐安装 CUDA 版本的 PyTorch（根据你的 CUDA 版本调整）
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

---

2. 验证安装

python3 -c "import flask; print('Flask 已安装')"
python3 -c "import torch; print(f'PyTorch 版本: {torch.__version__}, 是否使用GPU: {torch.cuda.is_available()}')"

---

3. 使用虚拟环境（推荐）

# 创建并激活虚拟环境
python3 -m venv myenv
source myenv/bin/activate

# 安装依赖
pip install flask sentence-transformers torch

# 运行脚本
python3 javaRunGpu.py --api  # 或命令行模式

---

4. 重新运行脚本

# API 模式
python3 javaRunGpu.py --api

# CLI 模式
python3 javaRunGpu.py "需要向量化的文本"

---

常见问题排查

问题现象	解决方案
No module named 'flask'	确保已正确执行 pip install flask
GPU 未被识别	检查 torch.cuda.is_available() 输出
端口冲突	使用 --port 指定其他端口
内存不足	添加 --api --host 127.0.0.1 限制外部访问