FLASK和GPU依赖安装
目录
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/cu1182. 验证安装
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 限制外部访问 |