加载预训练模型报错
收藏
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
收藏
请登录后评论
貌似是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
谢谢,早上我按照paddle首页的pip安装方式重装了一次paddle,就不报错了