Pytorch cuDNN错误:cuDNN_状态_执行_在使用flair时失败

Pytorch cuDNN错误:cuDNN_状态_执行_在使用flair时失败,pytorch,Pytorch,我一直在使用 尝试使用flair来测试flair,但我不知道为什么我不能使用GPU。 并尝试了以下方法: >>> from flair.data import Sentence >>> from flair.models import SequenceTagger >>> sentence = Sentence('I love Berlin .') >>> tagger = SequenceTagger.load('ner'

我一直在使用

尝试使用flair来测试flair,但我不知道为什么我不能使用GPU。 并尝试了以下方法:

>>> 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
>>来自flair.data导入语句
>>>从flair.models导入SequenceTagger
>>>句子=句子(‘我爱柏林’)
>>>tagger=SequenceTagger.load('ner')
2019-07-20 17:52:15062加载文件/home/vz/.flair/models/en-ner-conll03-v0.4.pt
回溯(最近一次呼叫最后一次):
文件“”,第1行,在
文件“/home/vz/miniconda3/envs/gp/lib/python3.6/site packages/flair/nn.py”,第103行,已加载
model=cls.\u init\u model\u与\u state\u dict(状态)
文件“/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/flair/models/sequence\u-tagger\u-model.py”,第205行,在初始模型中,带有
锁定的\u辍学=使用锁定的\u辍学,
文件“/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/flair/models/sequence\u-tagger\u-model.py”,第166行,在初始化中__
自我激励(天赋设备)
文件“/home/vz/miniconda3/envs/gp/lib/python3.6/site-packages/torch/nn/modules/module.py”,第386行,插入至
返回自我。\u应用(转换)
文件“/home/vz/miniconda3/envs/gp/lib/python3.6/site packages/torch/nn/modules/module.py”,第193行,适用于
模块应用(fn)
文件“/home/vz/miniconda3/envs/gp/lib/python3.6/site packages/torch/nn/modules/module.py”,第193行,适用于
模块应用(fn)
文件“/home/vz/miniconda3/envs/gp/lib/python3.6/site packages/torch/nn/modules/module.py”,第193行,适用于
模块应用(fn)
[上一行重复1次]
文件“/home/vz/miniconda3/envs/gp/lib/python3.6/site packages/torch/nn/modules/rnn.py”,第127行,适用于
自展平_参数()
文件“/home/vz/miniconda3/envs/gp/lib/python3.6/site packages/torch/nn/modules/rnn.py”,第123行,扁平化参数
self.batch_first,bool(self.directional))
运行时错误:cuDNN错误:cuDNN\u状态\u执行\u失败
有人能帮我解决这个错误吗?
提前感谢。

错误在于我的机器和CUDNN要求,我建议每个人都使用conda安装pytorch,所以安装方式应该是这样的


conda install pytorch torchvision cudatoolkit=9.0 -c pytorch


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

您可以在(Install PyTorch)中找到适用于您的平台的正确说明