search

luopeixiang / named_entity_recognition

中文命名实体识别（包括多种模型：HMM，CRF，BiLSTM，BiLSTM+CRF的具体实现）
2.12k stars 538 forks source link

cuDNN error: CUDNN_STATUS_EXECUTION_FAILED #52

Open luogyu7 opened 9 months ago

luogyu7 commented 9 months ago

正在训练评估双向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 版本 一直抱上述错误，有遇到同样错误的人吗？