禁用GPU设置
# 在import tensorflow之前import osos.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速度对比)就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。
最新文章
热门文章
猜你喜欢
