[问题]:调用paddlenlp的uie模型时,报错Segmentation fault (core dumped)

xlpyo6sf  于 2个月前  发布在  其他
关注(0)|答案(4)|浏览(47)

请提出你的问题:

使用脚本:

from paddlenlp import Taskflow
schema = ["成绩"]
ie = Taskflow('information_extraction', schema=schema)
allStr = "小明是个优秀的学生,这次考试成绩是100分"
res = ie(allStr)
data = {}
for key in schema:
    data[key] = res[0].get(key, [{'text': ''}])[0]['text']
print(data)

版本:

paddlepaddle-gpu=2.3.2.post112
paddlenlp=2.5.2
scipy=1.7.3
python==3.7

报错信息:

Segmentation fault (core dumped)
polkgigr

polkgigr1#

可能存在的原因如下:

  1. nvidia驱动过高,尝试降低驱动版本至cuda11.2对应驱动。
  2. cuda和cudnn没有正确安装,检查/usr/local/cuda/下是否存在cudnn。
    如果以上尝试无效,建议通过docker启动,可以通过 https://hub.docker.com/r/paddlepaddle/paddle/tags/?page=&page_size=&ordering=&name=11.2 进行查询合适的docker镜像。
41ik7eoe

41ik7eoe2#

驱动程序API版本:12.2,运行时API版本:11.2,cuDNN版本:8.1。这是我的版本,应该没问题,我之前在这个环境下运行paddleocr都没有问题。

czq61nw1

czq61nw13#

驱动程序API版本:12.2,运行时API版本:11.2,cuDNN版本:8.1。这是我的版本,应该没问题,我之前在这个环境下运行paddleocr都没有问题。
驱动程序版本可能过高,是否有尝试在import paddle上进行测试?

相关问题