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 限制外部访问 |