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则记录一些日志。

posted @ 2018-01-16 16:49  唐淼  阅读(1471)  评论(0编辑  收藏  举报