tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device for operation ‘model1_2/Softmax’: Could not satisfy explicit device specification ‘/device:GPU:0’ because no supported kernel for GPU devices is available.
今天再跑cleverhans代码的时候出现这个问题,也是很神奇啊,代码在笔记本上跑就没问题,咋在服务器上就不行了啊。后来我就看到别的博主然后就在tf.Session()的前面加上了1行代码:
config = tf.ConfigProto(allow_soft_placement=True)
再把原来的 sess = tf.Session()改为
sess = tf.Session(config=config)
它就好了,我看到博主的评论里有人说这还是用CPU跑的代码,经过我的观察,这资源管理器里面明显增加的是cuda啊,还有GPU的一些其他的东西飙升,显然还是用的GPU跑的。
更神奇的来了,我本来想在写博客的时候重现一些错误的我就恢复了原始代码的原始面貌,它居然还可以正确运行了,这到底是为啥啊!!!!目测是一个初始化的问题,感觉是代码找不到运行它的东西了,然后我修改了代码它就找到了,我再删除代码它可能就记住了就是那个玩意它可以用,然后就默认了,然后就可顺利跑了,先占个坑,以后有更深的理解在解释,继续做实验.
Could not satisfy explicit device specification ‘/device:GPU:0’ because no supported kernel for GPU
最新推荐文章于 2021-04-26 21:36:57 发布