[Stepping on the pit] Error: CUDNN_STATUS_EXECUTION_FAILED

Problem: After the model is loaded and starts training for the first epoch, an error CUDNN_STATUS_EXECUTION_FAILED is reported.

Reason: The versions of CUDA and Pytorch do not correspond.

Solution: Find the corresponding CUDA and Pytorch versions on the Pytorch official website and install them.

Guess you like

Origin blog.csdn.net/fuss1207/article/details/123158302