如何避免CUDNN_STATUS_NOT_INITIALIZED错误的发生?
时间: 2023-11-13 15:05:53 浏览: 111
Windows系统下配置深度学习Darknet框架使用YOLO神经网络算法实现目标检测(Windows10+VS2015+OpenCV3.4.0+CUDA9.0+cuDNN7.0)-附件资源
避免CUDNN_STATUS_NOT_INITIALIZED错误的发生,可以从以下几个方面入手:
1. 确认cuda、cudnn、显卡驱动、torch版本是否匹配;
2. 确认程序是否正确地初始化了cudnn;
3. 确认程序是否正确地释放了cudnn资源;
4. 确认程序是否正确地使用了cudnn API。
如果以上几个方面都没有问题,可以尝试重启程序或者重启电脑来解决该问题。
阅读全文