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
投稿猜你喜欢
- create table [order] ( code varchar(50), createtime datetime ) --应用 us
Vue3-KeepAlive,多个页面使用keepaliveVue3的生命周期未添加keepAlive的生命周期beforeCreate=&- 1.设置小数位数1.1 数据框设置统一小数位数以保留小数点后两位小数为例：import pandas as pdimport numpy a
- 一、Tesseract简介Tesseract是一个OCR库(OCR是英文Optical Character Recognition的缩写)，
- ASP使用xmlhttp获取远程网页内容，解决乱码问题方法一：<％function getHTTPPage(url)on error
- 本文实例讲述了Javascript连接Access数据库的方法。分享给大家供大家参考。具体实现方法如下：var roc = roc || {
- 本文默认你的开发环境.数据库已经安装好想用使用数据库.我们需要现在pom文件中添加相应的依赖pom.xml<?xml version=
- Python单元测试unittest中提供了一下四种装饰器实现测试跳过和预期故障。（使用Python 2.7.13）请查考Python手册中
- 最近在使用webpack + vue做个人娱乐项目时，发现npm run build后，css js img静态资源文件均找不到路径，报40
- 前台页面 $.ajax({ type: "post", contentType: "application/j
- 自动发送邮件能应用于许多场景，比如我想要知道股票策略中的股票池是否有实时的更新，这时候如果再拉一遍数据，跑一遍脚本，实在是太浪费时间了。为什
- 现在假如要写一个按照"标题",'内容','作者'等等进行针对性的选择,这时需要涉及到使用
- 本文实例为大家分享了python opencv摄像头应用的具体代码，供大家参考，具体内容如下1、安装下载安装包pip install ope
- 要随机生成字符串代码如下：在MySQL中定义一个随机串的方法，然后再SQL语句中调用此方法。随机串函数定义方法：CREATE DEFINER
- 脚本1：这两天用python写了一个删除指定目录下过期时间的脚本。也可能是我初学python，对python还不够熟习，总觉得这个脚本用sh
- 无图形界面的代码编写python代码作为脚本语言，其实最好直接使用vim在字符界面里面直接编写，这才是正道，但是作初学者，其实直接在服务器上
- class test { &nbs
- 微博模拟登录这是本次爬取的网址：https://weibo.com/一、请求分析找到登录的位置，填写用户名密码进行登录操作看看这次请求响应的
- 1.创建虚拟环境首先创建一个新文件夹在PyCharm终端中切换到这个文件夹，输入 python -m venv 环境名 创建虚拟环境&nbs
- 本文实例讲述了Python通用循环的构造方法。分享给大家供大家参考，具体如下：1.交互循环是无限循环的一种，允许用户通过交互的方式程序的特定
