美文网首页
Tensorflow常用会话设置GPU&ConfigProto

Tensorflow常用会话设置GPU&ConfigProto

作者: 5122爱你所爱 | 来源:发表于2019-06-17 20:58 被阅读0次

深度学习中常用的会话设置,仅供参考!

模块ConfigProto

创建session时常用的配置参数,使用tf.ConfigProto进行参数配置

with tf.Graph().as_default():

    gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.4)

    sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options,log_device_placement=True))

    with sess.as_default():

        print "example"

tf.ConfigProto()的参数

log_device_placement=True 是否打印设备分配日志

allow_soft_placement=True 如果指定的设备不存在,允许TF自动分配设备

gpu_options = tf.GPUOptions() 控制GPU资源

tf.GPUOptions() 参数

使用allow_growth参数,开始分配少量GPU容量,然后按需慢慢的增加,由于不释放内存,所以会导致碎片

tf.GPUOptions(allow_growth = True)

设置每个GPU应该拿出多少容量给进程使用

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.4)

GPU选择

方式一 CUDA_VISIBLE_DEVICES=0 python your.py  or   CUDA_VISIBLE_DEVICES=0,1 python your.py

方式二 os.environ['CUDA_VISIBLE_DEVICES'] = '0' #使用 GPU 0  or os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' # 使用 GPU 0,1

相关文章

  • Tensorflow常用会话设置GPU&ConfigProto

    深度学习中常用的会话设置,仅供参考! 模块ConfigProto 创建session时常用的配置参数,使用tf.C...

  • Tensorflow的基本使用

    使用tensorflow 你必须明白tensorflow 使用图(graph)来表示计算任务 在被称为会话 Ses...

  • Django-cookie&session

    会话跟踪 会话(Session)跟踪是Web程序中常用的技术,用来跟踪用户的整个会话。常用的会话跟踪技术是Cook...

  • tensorflow——Session 会话

    Session 会话 1. 功能 管理程序运行的资源,完成计算之后需要关闭会话来让系统回收资源。否则资源可能就不够...

  • Tensorflow之会话

    会话 创建 方法一 方法二 第一种方法在程序因为意外退出时,关闭会话的函数可能就不会被执行从而导致资源泄露,第二种...

  • 关于tensorflow 会话

    tf通过运行时维护的session本地线程栈,来管理默认session。故不同的线程会有不同的默认session,...

  • Tensorflow Session 配置选项解析

    在 Tensorflow 中需要通过创建 Session 类实例来为当前计算图注册会话,然后通过会话的 run 方...

  • 细读 JS | XSS、CSRF 浅谈

    一、前提 Cookie 常用于会话状态管理、个性化设置等。 在浏览器可以通过 document.cookie 来访...

  • iOS如何写Cookie

    前言 会话(Session)跟踪是Web程序中常用的技术,用来跟踪用户的整个会话。常用的会话跟踪技术是Cookie...

  • Tensorflow

    会话打开方式 神经网络参数与Tensorflow变量 在Tensorflow中,变量tf.Variable的作用就...

网友评论

      本文标题:Tensorflow常用会话设置GPU&ConfigProto

      本文链接:https://www.haomeiwen.com/subject/zgaafctx.html