[踩坑]报错：CUDNN_STATUS_EXECUTION_FAILED

问题：模型加载完成开始训练第一个epoch之后，报错CUDNN_STATUS_EXECUTION_FAILED。

原因：CUDA和Pytorch的版本不对应。

解决：在Pytorch官网中查找对应的CUDA和Pytorch的版本，进行安装。