TensorFlow深度学习设置显存占用率

最新推荐文章于 2024-07-17 21:20:07 发布
最新推荐文章于 2024-07-17 21:20:07 发布
阅读量2.3k 收藏 8
点赞数 2
CC 4.0 BY-SA版权
分类专栏： 深度学习 文章标签： tensorflow 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/half_lemon_c/article/details/109891628
本文介绍了如何查看GPU显存使用情况，并详细阐述了两种在TensorFlow中设置显存占用率的方法，包括动态分配和指定使用比例。通过在特定.py文件中插入代码，可以有效控制显存占用，例如从7888 MB降低到5420 MB，以优化深度学习训练过程。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

在使用TensorFlow深度学习的时候，会默认占用最大显存，如何设置显存占用率，总结如下：

I. 两种方法查看显存使用情况

  • 方法1：CMD窗口中输入命令nvidia-smi

在这里插入图片描述

  • 方法2： 使用软件AIDA64（推荐），实时监测电脑硬件使用情况

照片描述
电脑闲置可用显存为7443 MB。

II. 设置TensorFlow显存占用率

  • 方法1：按需动态分配显存（更灵活）
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
最低0.47元/天 解锁文章

200万优质内容无限畅学
解决tensorflow模型加载时把GPU显存占满的问题(亲测效果显著)
白羊是小白
08-27 1636
【代码】解决tensorflow模型加载时把GPU显存占满的问题(亲测效果显著)
TensorFlow使用GPU训练网络时多块显卡的显存使用问题
不眠旅行
06-22 1万+
使用实验室的服务器网络训练时，发现我只使用单个GPU跑程序，但三块显卡的显存都被占用。查了官网说明，发现这是因为TensorFlow训练时默认占用所有GPU的显存导致的。于是记录一下解决方法。
tensorflow_keras+GPU显存按需分配.doc
09-16
tensorflow_keras+GPU显存按需分配
tensorflow-gpu设置显存占用情况（Memory-Usage）
m0_55621747的博客
10-26 719
每次在服务器跑深度学习代码，我的数据量很小。但是内存直接占满。如下图可以设置内存占用情况。
Tensorflow显存占用情况
qq_43532776的博客
06-29 1475
Tensorflow 显存占用情况 在tensorflow中，如果不设置显存占用方式，那么默认会占用全部显存。这里将Tensorflow的显存使用方式设置为按需分配，观测其真实占用得而GPU显存资源情况，代码如下： # 代码参考 Tensorflow P219页 # 获取所有GPU设备列表 gpus = tf.config.experimental.list_physical_devices('GPU') if gpus: try: # 设置 GPU 显存占用为按需分配，增长式 for gpu in
tensorflowGPU显存占用
不正经的码狗
07-17 553
tf
tensorflow 1.13 (tf 1.13)中keras设置GPU显存大小
醉意流年go的博客
06-01 455
tensorflow 1.13 中keras设置GPU使用大小
TensorFlow2设置显存按需分配
追梦少年ML的博客
09-10 1194
在用TensorFlow进行深度学习时，容易遇到显存爆满的情况。 如下： numpy.core._exceptions.MemoryError: Unable to allocate 359. MiB for an array with shape (60000, 28, 28) and data type float64 解决方案就是设置TensorFlow2设置显存按需分配 gpus = tf.config.experimental.list_physical_devices(device_type=
Tensorflow设置GPU显存占用比例
酒酿小圆子呀～
03-17 1873
tensorflow在训练时默认占用所有GPU的显存。可以通过以下方式解决该问题： 1. 显式设定需要分配的显存比例 config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.6 session = tf.Session(config=config, ...) per_process_gpu_...
tensorflow设置显存自适应和显存比例.
threadroc的专栏
02-03 1545
转自:http://blog.youkuaiyun.com/cq361106306/article/details/52950081 用惯了theano.再用tensoflow发现一运行显存就满载了，吓得我吃了一个苹果。  用天朝搜索引擎毛都搜不到，于是翻墙找了下问题的解决方法，原来有两种  1. 按比例 config = tf.ConfigProto() config.gpu_options.per
tensorflow c接口设置内存占用
春江水暖鸭先知
12-25 277
https://github.com/tensorflow/tensorflow/issues/13853
tensorflow入门（二）---tensorflow显存管理
热门推荐
leibaojiangjun1的博客
12-15 1万+
在运行Tensorflow上面的blog的小程序的时候程序我们会遇到一个问题，当然这个问题不影响我们实际的结果计算，但是会给同样使用这台计算机的人带来麻烦，程序会自动调用所有能调用到的资源，并且全占满，在自己的PC上没问题，但是在服务器上，问题就很大，因为一旦你运行程序，占满显存别人就不能再用了，解决方法是我们通常是给程序运行指定一块GPU，然后限制其使用比例： import tensorflo
tensorflow中显存分配
weixin_44738399的博客
03-21 670
【代码】【无标题】
tensorflow2.0显存设置
qq_41422774的博客
12-22 4457
遇到一个问题： 新买显卡显存8G但是tensorflow运行的时候界面显示只有约6.3G的显存可用，如下图： 即限制了我的显存，具体原因为什么我也不知道，但原来的显存小一些的显卡就没有这个问题。 目前的解决办法是： 官方文档解决 然后对应的中文博客 总结一下，就是下面的两个办法： 设置每次使用时按需申请 # 获取所有GPU组成list gpus = tf.config.experimenta...
TensorFlow 显存使用机制
qq_29007291的博客
08-01 2272
默认情况下，TensorFlow 会映射进程可见的所有 GPU 的几乎所有 GPU 内存（取决于 CUDA_VISIBLE_DEVICES）。通过减少内存碎片，可以更有效地使用设备上相对宝贵的 GPU 内存资源。 在某些情况下，最理想的是进程只分配可用内存的一个子集，或者仅根据进程需要增加内存使用量。 TensorFlow 在 Session 上提供两个 Config 选项来进行控制。 (1)...
tensorflow 显存 训练_tensorflow设置最大GPU显存占比
weixin_39589693的博客
12-21 204
前言最近跑模型用服务器用的十分频繁，没注意把GPU显存全占完了，结果师兄的项目跑不起来了。愧对师兄，于是刚刚调整了一下tensorflow GPU训练显存的占比。贴一下方法，以后能用上。方法使用GPU加速训练时，tensorflow会默认申请分配所有的GPU资源，这就会导致同样使用显存的项目互相冲突跑不起来，所以可以做如下配置。其实我发现跑模型GPU占用70%的速度和100% emmmmm貌似真的...
Keras/Tensorflow(1.13) 加载模型后占用高显存解决
DongHappyyy
09-23 1951
Keras/Tensorflow(1.13) 加载模型后占用高显存解决
好玩的第一篇博客
qq_38279227的博客
07-15 220
test
tensorflow1、2设置不占满显存
yangchen1024的博客
09-06 1043
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
jupter跑深度学习模型gpu占用率一直为0
最新发布
03-12
好的，我现在需要解决用户在Jupyter Notebook中运行深度学习模型时GPU占用率为0的问题。首先，我需要回顾用户提供的引用内容，看看是否有相关的解决方案。 用户提到的引用[1]提到了elastic-jupyter-operator，这个...
评论 3
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值