PaddleNLP 管道qa任务调用uie模型报错Segmentation fault

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

环境配置:

paddlepaddle==2.4.1
paddlenlp==2.4.3
python=3.8

运行pipelines下的examples/unsupervised-question-answering/unsupervised_question_answering_example.py时,报错日志如下:

We are using <class 'paddlenlp.transformers.ernie.tokenizer.ErnieTokenizer'> to load '/home/gpu_user/.paddlenlp/pipelines/unsupervised_question_answering/uie-base-qa-filter'.
createing synthetic answers...
0%| | 0/2 [00:00<?, ?it/s]
Segmentation fault
qcuzuvrc

qcuzuvrc1#

使用GPU也存在相同的问题。

pod7payv

pod7payv2#

请考虑更新一下cuDNN版本。我使用uie-x模型时出现了Segmentation fault,我猜测是10.2版本的cudnn8引起的问题。在降级到cudnn7.6.5后,问题得到了解决。

t30tvxxf

t30tvxxf3#

考虑更新一下cudnn版本,我是用uie-x模型出现Segmentation fault,我粗大10.2,cudnn8,在降级到cudnn7.6.5后解决问题。
感谢~,试了下更新cudnn,现在已经跑通了

qvtsj1bj

qvtsj1bj4#

cuda cudnn这些版本 要自己慢慢试验吗

相关问题