named_entity_recognition icon indicating copy to clipboard operation
named_entity_recognition copied to clipboard

cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

Open luogyu7 opened this issue 1 year ago • 0 comments

正在训练评估双向LSTM模型... Traceback (most recent call last): File "main.py", line 73, in main() File "main.py", line 43, in main crf=False File "/home/l1/NER/named_entity_recognition/evaluate.py", line 64, in bilstm_train_and_eval bilstm_model = BILSTM_Model(vocab_size, out_size, crf=crf) File "/home/l1/NER/named_entity_recognition/models/bilstm_crf.py", line 31, in init self.hidden_size, out_size).to(self.device) File "/home/l1/anaconda3/envs/NER/lib/python3.7/site-packages/torch/nn/modules/module.py", line 381, in to return self._apply(convert) File "/home/l1/anaconda3/envs/NER/lib/python3.7/site-packages/torch/nn/modules/module.py", line 187, in _apply module._apply(fn) File "/home/l1/anaconda3/envs/NER/lib/python3.7/site-packages/torch/nn/modules/rnn.py", line 117, in _apply self.flatten_parameters() File "/home/l1/anaconda3/envs/NER/lib/python3.7/site-packages/torch/nn/modules/rnn.py", line 113, in flatten_parameters self.batch_first, bool(self.bidirectional)) RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

我按照requirements.txt安装了环境,使用python3.7 cuda10.0 cudnn v7.6.5 版本 一直抱上述错误,有遇到同样错误的人吗?

luogyu7 avatar Dec 05 '23 07:12 luogyu7