首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >cuDNN错误:使用flair时的CUDNN_STATUS_EXECUTION_FAILED

cuDNN错误:使用flair时的CUDNN_STATUS_EXECUTION_FAILED
EN

Stack Overflow用户
提问于 2019-07-20 12:28:15
回答 1查看 7.8K关注 0票数 0

我一直在使用https://github.com/zalandoresearch/flair#example-usage

尝试使用flair来实验flair,但是我不知道为什么我不能使用GPU。并尝试了如下:

代码语言:javascript
复制
>>> from flair.data import Sentence
>>> from flair.models import SequenceTagger
>>> sentence = Sentence('I love Berlin .')
>>> tagger = SequenceTagger.load('ner')
2019-07-20 17:52:15,062 loading file /home/vz/.flair/models/en-ner-conll03-v0.4.pt
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/flair/nn.py", line 103, in load
    model = cls._init_model_with_state_dict(state)
  File "/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/flair/models/sequence_tagger_model.py", line 205, in _init_model_with_state_dict
    locked_dropout=use_locked_dropout,
  File "/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/flair/models/sequence_tagger_model.py", line 166, in __init__
    self.to(flair.device)
  File "/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/torch/nn/modules/module.py", line 386, in to
    return self._apply(convert)
  File "/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/torch/nn/modules/module.py", line 193, in _apply
    module._apply(fn)
  File "/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/torch/nn/modules/module.py", line 193, in _apply
    module._apply(fn)
  File "/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/torch/nn/modules/module.py", line 193, in _apply
    module._apply(fn)
  [Previous line repeated 1 more time]
  File "/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/torch/nn/modules/rnn.py", line 127, in _apply
    self.flatten_parameters()
  File "/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/torch/nn/modules/rnn.py", line 123, in flatten_parameters
    self.batch_first, bool(self.bidirectional))
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

有人能帮我解决这个错误吗?提前谢谢。

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2019-07-24 18:57:17

这个错误与我的机器和CUDNN的要求有关,我建议每个人都用conda安装Py手电筒,所以安装的方法应该是这样的

代码语言:javascript
复制
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

以消除安装过程中的任何问题。

票数 4
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/57124735

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档