tensorflow使用CPU可以跑(运行),但是使用GPU却不能用的情况

Posted 唐淼

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tensorflow使用CPU可以跑(运行),但是使用GPU却不能用的情况相关的知识,希望对你有一定的参考价值。

在跑的时候可以让加些选项: with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True)): 其中allow_soft_placement能让tensorflow遇到无法用GPU跑的数据时,自动切换成CPU进行。 log_device_placement则记录一些日志。

以上是关于tensorflow使用CPU可以跑(运行),但是使用GPU却不能用的情况的主要内容,如果未能解决你的问题,请参考以下文章

gpu版本低能用啥跑代码

tensorflow只能在装有gpu的机器上运行

tensorflow能不能gpu cpu 一起训练

为啥tensorflow训练用GPU比CPU更慢了

tensorflow2添加命令使用cpu训练

tensorflow怎么删除gpu版本,下载cpu版本