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

[Daria8976]

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

[anthonyAndchen]

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

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

[Aidoul]

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

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