码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2018-01-17 01:14:30      阅读:525      评论:0      收藏:0      [点我收藏+]

标签:记录   gpo   切换   body   device   .config   选项   div   post   

在跑的时候可以让加些选项: 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却不能用的情况

标签:记录   gpo   切换   body   device   .config   选项   div   post   

原文地址:https://www.cnblogs.com/tangmiao/p/8297487.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!