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.