gpt4 book ai didi

使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)

转载 作者:qq735679552 更新时间:2022-09-29 22:32:09 27 4
gpt4 key购买 nike

CFSDN坚持开源创造价值,我们致力于搭建一个资源共享平台,让每一个IT人在这里找到属于你的精彩世界.

这篇CFSDN的博客文章使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)由作者收集整理,如果你对这篇文章有兴趣,记得点赞哟.

禁用GPU设置 。

?
1
2
3
# 在import tensorflow之前
import os
os.environ[ 'CUDA_VISIBLE_DEVICES' ] = '-1'

CPU与GPU对比 。

显卡:GTX 1066 。

CPU 。

使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)

GPU 。

使用Tensorflow-GPU禁用GPU设置(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速度对比)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持我.

原文链接:https://blog.csdn.net/c13232906050/article/details/83505563 。

最后此篇关于使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)的文章就讲到这里了,如果你想了解更多关于使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com