yolov3.weights转换成yolo.h5的问题

img


请问Check failed: err == cudaSuccess || err == cudaErrorInvalidValue Unexpected CUDA error: invalid argument这个是什么问题？怎么解决？

我大概记得这可能是因为cuda和cudnn版本与tensorflow不匹配。选择与tensorflow匹配的cuda和cudnn的正确版本可能会解决您的问题。

版本不一致导致的，你检查一下版本，或者卸载了重新安装一下。
还有一种解决方案就是 ，直接用cpu来进行转换