因为没有支持GPU设备的内核

时间:2018-06-06 17:51:38

标签: kernel gpu device assign

我自己做yolo v2。当我尝试训练时,我收到了错误消息:

  

因为没有支持GPU设备的内核。

我之前已经多次使用过GPU gtx1080ti所以我不知道为什么我的项目会收到这条消息。

我的源代码在我的github上:LINK

这就是我得到的信息

  

InvalidArgumentError:无法为操作'save / SaveV2'分配设备:无法满足显式设备规范'/ device:GPU:0',因为没有支持GPU设备的内核可用。   注册内核:     设备= 'CPU'

我知道在stackoverflow上有很多天才在深度学习。

请阅读我的问题!! :)

0 个答案:

没有答案