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.
在这里插入图片描述

在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()
Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐