search

FangShancheng / ABINet

Read Like Humans: Autonomous, Bidirectional and Iterative Language Modeling for Scene Text Recognition
Other
436 stars 73 forks source link

我在运行ABINet的时候遇到RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED #98

Open Daria8976 opened 1 year ago

Daria8976 commented 1 year ago

我是用单个3060在训练，命令为CUDA_VISIBLE_DEVICES=0 python main.py --config=configs/pretrain_vision_model.yaml，但是报错RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

anthonyAndchen commented 5 months ago

我是用单个3060在训练，命令为CUDA_VISIBLE_DEVICES=0 python main.py --config=configs/pretrain_vision_model.yaml，但是报错RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

你好，请问你解决这个问题了吗？

Aidoul commented 4 months ago

我是用单个3060在训练，命令为CUDA_VISIBLE_DEVICES=0 python main.py --config=configs/pretrain_vision_model.yaml，但是报错RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

想必你是按照作者的配置要求配置的环境吧，我也遇到了同样的问题，安装了torch1.8.0之后便没有此错误了