问题:模型加载完成开始训练第一个epoch之后,报错CUDNN_STATUS_EXECUTION_FAILED。 原因:CUDA和Pytorch的版本不对应。 解决:在Pytorch官网中查找对应的CUDA和Pytorch的版本,进行安装。