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