RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED You can try to repro this exception using t

搜到的可能的解决方法：

1、CUDA、Cudnn、pytorch、python、VS的版本问题，但我在图像中可以运行成功，所以版本是没问题的。
2、GPU内存不足，训练的话可以降低num_workers、batch，但我是推理，也不存在这个问题。
3、也可能是你的电脑不支持默认参数的多进程处理，要么在def parse_args()中找到多进程参数，设置为0，(也可能不在命令参数里，自己找一下)或者把原来的参数降低，要么就不使用GPU处理器；参考教程：https://blog.csdn.net/qq_44001342/article/details/115212449
4、在循环的最后添加如下一行代码，本来我在循环的第一行添加这个代码，但不知道为啥没有用。

torch.cuda.empty_cache()

解决方案

我用的方案3改小了num workers，有效解决！

参考博文：https://blog.csdn.net/yu_xinli/article/details/127084720