南强小屋 Design By 杰米
禁用GPU设置
# 在import tensorflow之前 import os os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
CPU与GPU对比
显卡:GTX 1066
CPU
GPU
简单测试:GPU比CPU快5秒
补充知识: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-GPU禁用GPU设置(CPU与GPU速度对比)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。
南强小屋 Design By 杰米
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
南强小屋 Design By 杰米
暂无使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)的评论...