使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)
时间：2021-01-24 11:08:55 | 栏目：Python代码 | 点击：次
禁用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则记录一些日志。