当前位置：首页 > article >正文

FLASK和GPU依赖安装

article 2025/3/17 11:09:40

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