相似度模型SimilarityModel部署指南

1 篇文章 0 订阅
1 篇文章 0 订阅

相似度模型服务

对输入的两个句子进行推理预测,获得它们的语义相似度

环境依赖

工具说明版本
python开发环境3.8.10
paddlepaddle深度学习框架2.0.2 cpu版
flaskpython web框架,将调用转换为http请求
geventpython 并发框架
protobuf高效的结构化数据存储格式3.20.0
numpy数值计算1.19.3
scipy基于 Numpy 的科学计算库1.7.0

环境说明

centos:7.6 x64
python:python3.8.10 ; 编译安装 ;安装目录/usr/local/python3.8.10
NRNIE:stable-1.0.1 ;解压目录/home/project/devops/ERNIE_1.0.1

环境搭建

一.Python环境安装(3.8.10)

根据实际运行的情况,最好先安装好以下软件

  • libffi-devel
    安装方式:yum install libffi-devel -y
    缺少可能造成的问题:ModuleNotFoundError: No module named ‘_ctypes‘错误
  • zlib*
    安装方式:yum -y install zlib*
    缺少可能造成的问题:这个不安装,编译可能会报错
  1. 下载python3.8.10版本到本地:
    • 切换目录
      cd /usr/local
      
    • 下载版本
      wget https://www.python.org/ftp/python/3.8.10/Python-3.8.10.tgz
      
    • 解压安装
      tar -zxvf  Python-3.8.10.tgz
      
    • 进入解压目录进行安装
      cd /usr/local/Python-3.8.10    #进入解压后的源码目录进行编译
      ./configure --prefix=/usr/local/python3.8.10   #安装后的目录
      make && make install  #开始编译,等待时间较长
      
  2. 设置软链接,验证版本
    • 设置软连接
      #将安装好的python3.8执行文件软链接到 /usr/bin/python3  
      sudo ln -s /usr/local/python3.8.10/bin/python3.8 /usr/bin/python3  
      #将安装好的pip执行文件软链接到 /usr/bin/pip3
      sudo ln -s /usr/local/python3.8.10/bin/pip3 /usr/bin/pip3
      
      如过提示软链接文件已存在,可以先执行删除原来的软链接:rm -rf /usr/bin/python3
    • 验证
      python3 --version
      
      输出3.8.10即为安装完成
  3. 安装其他python依赖库
    • flask & gevent

      python3 -m pip install flask gevent
      
    • protobuf

      python3 -m pip install protobuf==3.20.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
      
    • numpy

      python3 -m pip install numpy==1.19.3 -i https://pypi.tuna.tsinghua.edu.cn/simple
      
    • numpy

      python3 -m pip install scipy==1.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
      

      注:-i https://pypi.tuna.tsinghua.edu.cn/simple为补充参数,表示本次使用清华大学的源进行下载

二.深度学习环境paddlepadle

方式1. pip安装

# gpu版
python3 -m pip install paddlepaddle-gpu==2.0.2.post100 -f https://paddlepaddle.org.cn/whl/mkl/stable.html
# cpu版
python3 -m pip install paddlepaddle==2.0.2 -i https://mirror.baidu.com/pypi/simple

方式2. 官网安装

参考快速安装

三.语言模型下载

该项目使用深度语言模型ERNIE,并基于预训练模型进行微调

  1. 所需的预训练模型是ERNIE1.0中文Base模型,点击下载

  2. 下载后解压到预定目录,为后面conf配置引用做准备

    tar -zxvf ERNIE_stable-1.0.1.tar.gz
    

服务启动

配置参数

conf/model_conf.json配置了常见的模型参数,如果这些参数不够满足要求,可以根据实际需求更改,然后重写模型加载前的参数读取方法即可

下面是配置参数的示例:

{
  "ernie": {
    "use_cuda": false,
    "batch_size": 32,
    "init_checkpoint": "/home/project/devops/ERNIE_1.0.1/params",
    "ernie_config_path": "/home/project/devops/ERNIE_1.0.1/ernie_config.json",
    "vocab_path": "/home/project/devops/ERNIE_1.0.1/vocab.txt",
    "save_inference_model_path": "/home/project/devops/ERNIE_1.0.1/params"
  }
}
  • use_cuda表示是否使用GPU
  • init_checkpoint表示模型要加载的检查点(训练好的模型输出的变量、参数等数据)
  • ernie_config_pathvocab_path是预训练模型的参数和词典
  • save_inference_model_path为模型推理时加载的模型文件。

如果没有训练,可以直接使用预训练模型进行推理预测,即init_checkpoint的路径改为预训练模型的params即可(如示例)。

启动服务

进入该项目的根目录,然后执行命令:

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值