首页 > 其他 > 详细

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

时间:2018-01-16 20:15:31      阅读:1116      评论:0      收藏:0      [点我收藏+]

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

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!