paddlenlp加载模型报错
收藏
在『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
收藏
请登录后评论
请问下你自己机器的环境是?可能是环境不支持导致的问题
同样报错+1,求解决方案