I’m sorry, you’re right. With or without plotting part doesn’t affect the result, the cudnn_error still occurs. I think it’s just lucky that I can run the code normally two days ago. Now the error still exist.