我自己做yolo v2。当我尝试训练时,我收到了错误消息:
因为没有支持GPU设备的内核。
我之前已经多次使用过GPU gtx1080ti所以我不知道为什么我的项目会收到这条消息。
我的源代码在我的github上:LINK
这就是我得到的信息
InvalidArgumentError:无法为操作'save / SaveV2'分配设备:无法满足显式设备规范'/ device:GPU:0',因为没有支持GPU设备的内核可用。 注册内核: 设备= 'CPU'
我知道在stackoverflow上有很多天才在深度学习。
请阅读我的问题!! :)