在跑的时候可以让加些选项: 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却不能用的情况
Posted 唐淼
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tensorflow使用CPU可以跑(运行),但是使用GPU却不能用的情况相关的知识,希望对你有一定的参考价值。
以上是关于tensorflow使用CPU可以跑(运行),但是使用GPU却不能用的情况的主要内容,如果未能解决你的问题,请参考以下文章