Tensorflow-CPU与禁用GPU设置的对比有什么不同(cpu,gpu,tensorflow,开发技术)

时间:2024-05-03 03:47:15 作者 : 石家庄SEO 分类 : 开发技术
  • TAG :

禁用GPU设置

# 在import tensorflow之前import osos.environ['CUDA_VISIBLE_DEVICES'] = '-1'

CPU与GPU对比

显卡:GTX 1066

CPU

Tensorflow-CPU与禁用GPU设置的对比有什么不同

GPU

Tensorflow-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则记录一些日志。

 </div> <div class="zixun-tj-product adv-bottom"></div> </div> </div> <div class="prve-next-news">
本文:Tensorflow-CPU与禁用GPU设置的对比有什么不同的详细内容,希望对您有所帮助,信息来源于网络。
上一篇:关于Jdbctemplate多数据源如何配置的方法下一篇:

15 人围观 / 0 条评论 ↓快速评论↓

(必须)

(必须,保密)

阿狸1 阿狸2 阿狸3 阿狸4 阿狸5 阿狸6 阿狸7 阿狸8 阿狸9 阿狸10 阿狸11 阿狸12 阿狸13 阿狸14 阿狸15 阿狸16 阿狸17 阿狸18