首页 Paddle框架 帖子详情
paddlenlp加载模型报错
收藏
快速回复
Paddle框架 问答模型训练 2234 2
paddlenlp加载模型报错
收藏
快速回复
Paddle框架 问答模型训练 2234 2

在『2021语言与智能技术竞赛』- 事件抽取任务基线系统 的基线NoteBook中,下载代码到自己机器上运行,运行起来特别慢,而且执行代码时报错:parallel_for failed: cudaErrorNoKernelImageForDevice: no kernel image is available for execution on the device. 有没有大神解释一下什么原因

    
from paddlenlp.transformers import ErnieForTokenClassification, ErnieForSequenceClassification
from utils import load_dict

label_map = load_dict('./conf/DuEE-Fin/trigger_tag.dict')
id2label = {val: key for key, val in label_map.items()}
print(id2label)
# model = ErnieForTokenClassification.from_pretrained("ernie-1.0", num_classes=len(label_map))


# from paddlenlp.transformers import ErnieForSequenceClassification

model = ErnieForSequenceClassification.from_pretrained("ernie-1.0", num_classes=len(label_map))

line13 为最后一行,完整报错:

W0509 13:09:59.510846 14922 device_context.cc:362] Please NOTE: device: 0, GPU Compute Capability: 8.6, Driver API Version: 11.1, Runtime API Version: 10.2
W0509 13:09:59.515894 14922 device_context.cc:372] device: 0, cuDNN Version: 7.6.
Traceback (most recent call last):
File "dueStep2.py", line 13, in 
model = ErnieForSequenceClassification.from_pretrained("ernie-1.0", num_classes=len(label_map))
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddlenlp/transformers/model_utils.py", line 229, in from_pretrained
if k in base_parameters_dict:
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddlenlp/transformers/utils.py", line 83, in __impl__
init_func(self, *args, **kwargs)
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddlenlp/transformers/ernie/modeling.py", line 203, in __init__
self.pad_token_id = pad_token_id
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddlenlp/transformers/ernie/modeling.py", line 41, in __init__
self.word_embeddings = nn.Embedding(
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddle/nn/layer/common.py", line 1348, in __init__
self.weight = self.create_parameter(
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddle/fluid/dygraph/layers.py", line 407, in create_parameter
return self._helper.create_parameter(temp_attr, shape, dtype, is_bias,
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddle/fluid/layer_helper_base.py", line 367, in create_parameter
return self.main_program.global_block().create_parameter(
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddle/fluid/framework.py", line 2988, in create_parameter
initializer(param, self)
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddle/fluid/initializer.py", line 557, in __call__
op = block._prepend_op(
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddle/fluid/framework.py", line 3100, in _prepend_op
_dygraph_tracer().trace_op(type,
File "/home/XXX/miniconda3/lib/python3.8/site-packages/paddle/fluid/dygraph/tracer.py", line 43, in trace_op
self.trace(type, inputs, outputs, attrs,
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:172)
0
收藏
回复
全部评论(2)
时间顺序
TC.Long
#2 回复于2021-06

请问下你自己机器的环境是?可能是环境不支持导致的问题

0
回复
z
zonlinger
#3 回复于2021-06

同样报错+1,求解决方案

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