首页 PaddleNLP 帖子详情
加载预训练模型报错
收藏
快速回复
PaddleNLP 其他NLP训练 2122 2
加载预训练模型报错
收藏
快速回复
PaddleNLP 其他NLP训练 2122 2

paddle(GPU)和paddlenlp都升级到最新版本了,

加载预训练模型,trial = paddlenlp.transformers.ErnieGramModel.from_pretrained('ernie-gram-zh')
出现如下提示:

SystemError: (Fatal) Operator uniform_random raises an thrust::system::system_error exception.
The exception content is
:parallel_for failed: cudaErrorNoKernelImageForDevice: no kernel image is available for execution on the device. (at /paddle/paddle/fluid/imperative/tracer.cc:221)

0
收藏
回复
全部评论(2)
时间顺序
DeepGeGe
#2 回复于2022-01

貌似是CUDA版本问题,在GitHub上有链接说了这个问题:

https://github.com/PaddlePaddle/Paddle/issues/35008

https://github.com/PaddlePaddle/Paddle/issues/37134

还有这里有小伙伴说安装2.0..1版本可以:

https://github.com/PaddlePaddle/Paddle/issues/32199

0
回复
johnyanccer
#3 回复于2022-01

谢谢,早上我按照paddle首页的pip安装方式重装了一次paddle,就不报错了

1
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户