tensorflow 限制显存大小的实现
作者：安阳小栈-客官歇会吧 发布时间：2023-03-04 02:19:59
标签：tensorflow,限制,显存
Python在用GPU跑模型的时候最好开多进程，因为很明显这种任务就是计算密集型的。
用进程池好管理，但是tensorflow默认情况会最大占用显存，尽管该任务并不需要这么多，因此我们可以设置显存的按需获取，这样程序就不会死掉了。
1. 按比例预留：
tf_config = tensorflow.ConfigProto()
tf_config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 分配50％
session = tensorflow.Session(config=tf_config)
2. 或者干脆自适应然后自动增长：
tf_config = tensorflow.ConfigProto()
tf_config.gpu_options.allow_growth = True # 自适应
session = tensorflow.Session(config=tf_config)
来源：https://www.cnblogs.com/peter1994/p/7834137.html
0
投稿猜你喜欢
- 内容概要：print() 是一个常用函数。那么，您是否注意过，print() 会在显示当前语句后换行。如果遇到需要连续显示、不换行的情况，比
pytorch定义新的自动求导函数在pytorch中想自定义求导函数，通过实现torch.autograd.Function并重写forwa- 钉钉设置机器人首先在钉钉设置钉钉机器人群设置—> 智能群助手—>添加机器人—>自定义添加完成，得到一个Webhook AP
- 这可是个综合性的问题，看看下面对文件操作的集大成代码：<％ 'Set file i/
- 本文实例为大家分享了python实现邮箱发送信息的具体代码，供大家参考，具体内容如下一、SSLSSL 是指安全套接字层，简而言之，它是一项标
- 1.在html页面中导入js文件和css文件<link rel="stylesheet" href=".
- Oracle关系型数据库管理系统是世界上流行的关系数据库，它是一个极其强大、灵活和复杂的系统，据说，在使用oracle时应有这样的思想，那就
- 一、object类的源码class object: """ The most bas
- 以一种有意义的方式组织数据可能是一项挑战。有时你需要的可能是一个简单的排序，但是通常你需要做更多，你需要分组来进行分析和统计。幸运的是，SQ
- 本文实例讲述了PHP实现对图片的反色处理功能。分享给大家供大家参考，具体如下：今天有个需求用php对图片进行反色，和转灰，之前不知道可不可行
- 用pytesseract识别图片中的数字Win 平台 使用步骤一、安装包。二、找个图片，运行如下识别程序。示例程序：import pytes
- requests是python的一个HTTP客户端库，跟urllib，urllib2类似，那为什么要用requests而不用urllib2呢
- 现在我主要教大家如何去实战，做一个简易的知乎日报API 首先你要熟悉django的基本用法，会写模型，会写视图函数，会配置url。1.配置字
- PHP mysqli_sqlstate() 函数返回最后一个 MySQL 操作的 SQLSTATE 错误代码：<?php// 假定数据
- 这篇文章主要介绍了Python如何基于smtplib发不同格式的邮件,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习
- Oracle公司6月9日宣布同意收购TimesTen公司，TimesTen是一家私营软件企业，其产品能提高用于股市和机票预订等需要快速响应时
- 一. 分析需求1. 需求说明在项目开发过程中,我们有时会频繁的更新代码, 流程大概为:(1) 本地git push提交代码至git托管平台(
- 在本地使用pycharm连接databricks，大致步骤如下：首先，为了让本地环境能够识别远端的databricks集群环境，需要收集da
- 制作这个播放器的目的是为了将下载下来的mp3文件进行随机或是顺序的播放。选择需要播放的音乐的路径，选择播放方式，经过测试可以完美的播放本地音
- xhtml+css页面制作过程中问题的解决方案，说是解决方案应该有点过了，充其量只不过是给刚刚开始学标准页面制作的朋友们的一些小建议，如果讲
