TensorFlow限制占用显存

最新推荐文章于 2022-03-15 09:39:16 发布
最新推荐文章于 2022-03-15 09:39:16 发布
阅读量289 收藏 1
点赞数
分类专栏： 使用命令
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/sinat_41789374/article/details/103480614
版权

不设置的话会把剩下所有的显存都占满

tf_config = tf.ConfigProto()  
tf_config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 分配50%  
tf_config.gpu_options.allow_growth = True # 自适应 （按照设置的batch_size占用显存）
session = tf.Session(config=tf_config)  #或者使用with tf.Session(config=config) as sess:
sinat_41789374
关注
专栏目录
tensorflow1、2设置不占满显存
yangchen1024的博客
09-06 959
Preface tensorflow的默认调度策略是吃满显存，如果我们模型较小、显存较大、占用率不足40%（自己估计的）时，是可以利用GPU同时训练两个模型的。 tensorflow 1 config = tf.ConfigProto() config.gpu_options.allow_growth = True sess = tf.Session(config=config) tensorflow 2 gpus = tf.config.list_physical_devices(device_type
Tensorflow设置显存自适应，显存比例
tsq292978891的博客
12-07 4485
Tensorfow框架下，在模型运行时，设置对显存的占用。 1. 按比例 config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.4 # 根据自己的需求确定 session = tf.Session(config=config, ...) 2. 自适应 config
深度学习笔记：windows+tensorflow 指定GPU占用内存（解决gpu爆炸问题）
去向前方的博客
07-05 1万+
目录 目录 前言 一、指定显卡 二、限制GPU用量 1、设置使用GPU的百分比 三、指定GPU并且限制GPU用量 指定第一块GPU可用 前言 最近在使用深度学习，跑了一个大的model，然后GPU炸了，上网搜索了一下如何解决这个问题，做下笔记，分享给大家。 keras在使用GPU的时候有个特点，就是默认全部占满显存。 这样如果有多个模型都需要使用GPU跑的话，那么限制是很大...
tensorflow如何避免内存泄漏
qq_38261075的博客
04-10 266
主要可以看看一些循环中使用的 tf 函数 可参考https://blog.csdn.net/qq_25737169/article/details/78125550?utm_source=blogxgwz8 https://cloud.tencent.com/developer/article/1599930 ...
解决Tensorflow显存溢出的问题
Ben出没冇注意的博客
01-04 1633
import tensorflow as tf import tensorflow.keras.backend as K os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true" config = tf.ConfigProto() config.gpu_options.allow_growth = True sess = tf.Session(config=config) K.set_session(sess) 加入上面几.
解决Tensorflow占用GPU显存问题
09-18
今天小编就为大家分享一篇解决Tensorflow占用GPU显存问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
tensorflow 限制显存大小的实现
12-20
为了解决这个问题，可以通过配置TensorFlow会话（Session）来限制GPU显存的使用。 1. **按比例预留显存**： 在TensorFlow中，可以使用`ConfigProto`类来设置GPU选项。通过设置`gpu_options.per_process_gpu_memory...
tensorflow_keras+GPU显存按需分配.doc
09-16
这个功能允许TensorFlow在运行过程中逐渐增加GPU的内存分配，而不是一开始就分配所有可用的显存。这可以防止因初始分配过大而导致的浪费。示例代码如下： ```python import tensorflow as tf physical_devices ...
tensorflow2总是爆显存的问题
不平凡之路
03-12 5511
最近在研究tensorflow2，在win环境下装了gpu版本，效率是高很多，但是出现几次爆显存的情况，就是显存被占满了，甚至机子直接黑屏的情况，查了一下资料，tensorflow默认是分配全部显存的，所以会导致显存爆掉。当然可以通过设置来调整： import os os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true" 实测，设置之后确实不存在爆显...
Tensorflow设置显存自适应,显存比例的操作
09-18
今天小编就为大家分享一篇Tensorflow设置显存自适应,显存比例的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
tensorflow基本版本编程问题和显存爆掉问题记录
zazazaz1的博客
08-18 642
AttributeError: module ‘tensorflow’ has no attribute 'Session’错误解决 import tensorflow as tf tf.Session() 报错 错误原因：报错AttributeError: module ‘tensorflow’ has no attribute ‘Session’。这是因为编程采用的是1.0的写法，但是在新的Tensorflow 2.0版本中已经移除了Session这一模块，解决办法有两个 1:改换运行代码 tf.c.
TensorFlow:显卡的指定和显存的控制
你悄悄的来到了我的博客
06-28 479
https://blog.csdn.net/m0_37041325/article/details/77488981
Tensorflow 2.0 GPU显存爆炸打问题及释放内存打方法
楠木大哥技术博客
08-16 8863
GPU 显存爆炸问题 训练过程中及其容易将显存占满，需要限制显存打使用。 import tensorflow as tf gpus = tf.config.experimental.list_physical_devices('GPU') if gpus: tf.config.experimental.set_virtual_device_configuration(gpus[0],[tf.config.experimental.VirtualDeviceConfiguration(memory_l
TensorFlow限制占用显存比例以及实现占用显存自增长
想写就写咯
04-26 3267
当实验室的服务器紧张，一张卡要几个人用时，就得预留显存给别人了，如果默认模式，TensorFlow会占用所有的freememory，这就不友好了。除此之外，如果指定显存自增长，也方便你知道自己的程序需要多大的显存。 tf_config = tf.ConfigProto() tf_config.gpu_options.per_process_gpu_memory_fraction = 0.5 #...
使用gpu版的tensorflow时出现OOM错误(显存溢出错误)： CUDA out of memory的解决办法。
m0_56192771的博客
03-15 5095
gpu版的tensorflow 在执行过程中会默认使用全部的 GPU 内存，给系统保留 200 M，但是在我的系统上会在分配内存时被拒绝导致报错，因此我们可以使用如下语句指定 GPU 内存的分配空间： 下面以我自己的电脑为例，进行相关设置。因为我的电脑的显存是6G，所以我可以设置tensorflow程序运行时所占用的显存不能超过6G，这里限制在4G大小。 import tensorflow as tf #限制消耗固定大小的显存（程序不会超出限定的显存大小，若超出的报错）。 physical_gpus = t
tensorflow显存管理、tensorflow使用多个gpu训练
热门推荐
liulina603的专栏
05-03 3万+
 通常在程序开始之前并不知道需要多大的显存，程序会去申请GPU的显存的50% 比如一个8G的内存，被占用了2G，那么程序会申请4G的显存（因为有足够的剩余显存） 如果此时显存被占用7G，那么程序会申请剩下的所有的1G的显存。也许你的程序根本用不着这么多显存，200M也许就够了，这时候如果程序能按照需求去申请就好了，幸运的是，这样的方法是存在的： import tensorflow as t...
TensorFlow学习 理解batchsize及其大小的影响
qq_36588760的博客
04-02 5276
之前面试过程中被问到过两个问题： （1）深度学习中batch size的大小对训练过程的影响是什么样的？ （2）有些时候不可避免地要用超大batch，比如人脸识别，可能每个batch要有几万甚至几十万张人脸图像，训练过程中超大batch有什么优缺点，如何尽可能地避免超大batch带来的负面影响？-------------------------------面试版回答----------...
tensorflow GPU显存控制
silent56_th的博客
11-06 1977
参考资料：https://www.tensorflow.org/tutorials/using_gpu 文中提到了两个控制方法：设置使用显存的百分比，以及设置逐渐增加使用的GPU显存。 关于第一种方法，可以参考Keras指定使用GPU、Keras设定GPU使用内存大小(Tensorflow backend)。 这里翻译第二种方法，代码如下：config = tf.ConfigProto() c
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值