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

 更新时间:2020年06月30日 09:08:48   转载 作者:ArchonGum  
这篇文章主要介绍了使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比),具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

禁用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速度对比)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • NumPy 数学函数及代数运算的实现代码

    NumPy 数学函数及代数运算的实现代码

    这篇文章主要介绍了NumPy 数学函数及代数运算的实现代码,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2018-07-07
  • Python安装OpenCV的示例代码

    Python安装OpenCV的示例代码

    这篇文章主要介绍了Python安装OpenCV的示例代码,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2020-03-03
  • 详解Python中列表和元祖的使用方法

    详解Python中列表和元祖的使用方法

    这篇文章主要介绍了详解Python中列表和元祖的使用方法,代码基于Python2.x版本,元祖和列表是Python学习当中的基础知识,需要的朋友可以参考下
    2015-04-04
  • pandas DataFrame数据转为list的方法

    pandas DataFrame数据转为list的方法

    下面小编就为大家分享一篇pandas DataFrame数据转为list的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-04-04
  • python hashlib加密实现代码

    python hashlib加密实现代码

    这篇文章主要介绍了python hashlib加密实现代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-10-10
  • Python 流程控制实例代码

    Python 流程控制实例代码

    Python是一门简单的语言。对于一个问题,应该只有一个解决方法。在Python中,有三种流程控制方法:if-else、while和for。
    2009-09-09
  • python中xrange用法分析

    python中xrange用法分析

    这篇文章主要介绍了python中xrange用法,实例分析了xrange的功能、作用以及与range在使用中的区别,具有一定参考借鉴价值,需要的朋友可以参考下
    2015-04-04
  • python矩阵转换为一维数组的实例

    python矩阵转换为一维数组的实例

    今天小编就为大家分享一篇python矩阵转换为一维数组的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-06-06
  • Python操作qml对象过程详解

    Python操作qml对象过程详解

    这篇文章主要介绍了Python操作qml对象过程详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-09-09
  • Python中Collections模块的Counter容器类使用教程

    Python中Collections模块的Counter容器类使用教程

    Counter是Python标准库提供的一个非常有用的容器,可以用来对序列中出现的各个元素进行计数,下面就来一起看一下Python中Collections模块的Counter容器类使用教程
    2016-05-05

最新评论