cuDNNerror:CUDNN_STATUS_EXECUTION_FAILED报错(已解决)

Python报错:

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
You can try to repro this exception using the following code snippet. If that doesn’t trigger the error, please include your original repro script when reporting this issue.
cuDNNerror:CUDNN_STATUS_EXECUTION_FAILED报错(已解决)_第1张图片

在DDRNet推理代码中,使用图片进行分割没有发现任何问题,但在接入双目相机视频流,再对视频的单帧图像进行分割,发现报如上错误。

网络上对这个问题的说明大多是:

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

在我的项目中,不使用GPU是不行的,视频分割帧率就完蛋了。
受以上启发,感觉我的问题也在于内存不足,尽管我在运行的过程中也观察了GPU使用率,不超过60%。。。

最后解决方案:

在循环的最后添加如下一行代码,本来我在循环的第一行添加这个代码,但不知道为啥没有用。

torch.cuda.empty_cache()

你可能感兴趣的:(报错处理,python,深度学习,开发语言)