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

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

http://www.kler.cn/a/586811.html

相关文章:

  • Linux驱动开发框架基础——新旧字符设备驱动笔记整理(1)
  • 软考中级设计模式实战宝典:核心意图解析+适用场景+完整源码下载+高频题库
  • 如何在AVL树中高效插入并保持平衡:一步步掌握旋转与平衡因子 —— 平衡因子以及AVL结构篇
  • 【mysql】centOS7安装mysql详细操作步骤!—通过tar包方式
  • 机器人技能列表
  • 通过特征值和特征向量实现的图像压缩和特征提取
  • 大模型训练全流程深度解析
  • 力扣-单调栈-739 每日温度
  • spring-aop笔记
  • 2023 年全国职业院校技能大赛(中职组)移动应用与开发赛项 赛题第十套
  • Python基础入门掌握(九)
  • 设计签到 用redis 和 MySQL
  • 目标检测中衡量模型速度和精度的指标:FPS和mAP
  • 软件测试之测试覆盖率
  • c++三级(枚举问题)
  • 普通enum和enum class的区别
  • Winfrom基础知识41-60
  • Uniapp 开发 App 端上架用户隐私协议实现指南
  • Halcon 和 opencv比有什么区别与优劣
  • python socket库详解