Tensorflow设置显存自适应,显存比例的操作


Posted in Python onFebruary 03, 2020

Tensorfow框架下,在模型运行时,设置对显存的占用。

1. 按比例

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4 # 根据自己的需求确定
session = tf.Session(config=config, ...)

2. 自适应

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)

设置GPU的使用率的时候,都是在创建Session的时候,对config类进行设置。

此外,当电脑上有多块GPU的时候,可以指定选取哪一快GPU进行计算。

# 在程序开头添加
os.environ['CUDA_VISIBLE_DEVICES'] = '0' #使用 GPU 0 0对应着ubuntu系统给GPU的序号,可通过Nvidia-smi命令查看

若存在多个GPU

os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' # 使用 GPU 0,1

一个常见的在代码中指定GPU使用的范例:

import tensorflow as tf
import numpy as np
import os

os.environ['CUDA_VISIBLE_DEVICES']='0'
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction=0.6

with tf.Session(graph=...,config=config) as sess:

## 后续的操作

以上这篇Tensorflow设置显存自适应,显存比例的操作就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python进阶教程之函数参数的多种传递方法
Aug 30 Python
修改Python的pyxmpp2中的主循环使其提高性能
Apr 24 Python
Python中自定义函数的教程
Apr 27 Python
利用Python如何批量修改数据库执行Sql文件
Jul 29 Python
Python实现将数据写入netCDF4中的方法示例
Aug 30 Python
在python中实现将一张图片剪切成四份的方法
Dec 05 Python
Python Pandas数据结构简单介绍
Jul 03 Python
Python获取时间范围内日期列表和周列表的函数
Aug 05 Python
Pytorch中实现只导入部分模型参数的方式
Jan 02 Python
pytorch程序异常后删除占用的显存操作
Jan 13 Python
python批量生成条形码的示例
Oct 10 Python
pandas求平均数和中位数的方法实例
Aug 04 Python
解决Tensorflow占用GPU显存问题
Feb 03 #Python
通过Turtle库在Python中绘制一个鼠年福鼠
Feb 03 #Python
python爬虫模块URL管理器模块用法解析
Feb 03 #Python
Tensorflow实现多GPU并行方式
Feb 03 #Python
python如何通过twisted搭建socket服务
Feb 03 #Python
关于Tensorflow分布式并行策略
Feb 03 #Python
基于python修改srt字幕的时间轴
Feb 03 #Python
You might like
PHP中在数据库中保存Checkbox数据(2)
2006/10/09 PHP
消息持续发送的完整例子
2006/10/09 PHP
PHP生成器简单实例
2015/05/13 PHP
PHP使用strtotime获取上个月、下个月、本月的日期
2015/12/30 PHP
PHP实现的文件上传类与用法详解
2017/07/05 PHP
一些主流JS框架中DOMReady事件的实现小结
2011/02/12 Javascript
基于jquery实现等比缩放图片
2014/12/03 Javascript
jquery带翻页动画的电子杂志代码分享
2015/08/21 Javascript
JavaScript入门系列之知识点总结
2016/03/24 Javascript
Bootstrap3制作自己的导航栏
2016/05/12 Javascript
Javascript 正则表达式校验数字的简单实例
2016/11/02 Javascript
原生js验证简洁注册登录页面
2016/12/17 Javascript
jquery+css实现简单的图片轮播效果
2017/08/07 jQuery
ECMAscript 变量作用域总结概括
2017/08/18 Javascript
微信小程序图片选择区域裁剪实现方法
2017/12/02 Javascript
Vue表单控件绑定图文详解
2019/02/11 Javascript
Vue全局loading及错误提示的思路与实现
2019/08/09 Javascript
使用layui的layer组件做弹出层的例子
2019/09/27 Javascript
Postman动态获取返回值过程详解
2020/06/30 Javascript
[10:14]2018DOTA2国际邀请赛寻真——paiN Gaming不仅为自己而战
2018/08/14 DOTA
[47:35]VP vs Pain 2018国际邀请赛小组赛BO2 第一场 8.18
2018/08/20 DOTA
编程语言Python的发展史
2014/09/26 Python
python自定义解析简单xml格式文件的方法
2015/05/11 Python
Python 数据结构之队列的实现
2017/01/22 Python
django之session与分页(实例讲解)
2017/11/13 Python
Python应用库大全总结
2018/05/30 Python
Python实现把类当做字典来访问
2019/12/16 Python
护理工作感言
2014/01/16 职场文书
小学教师管理制度
2014/01/18 职场文书
学生会竞选演讲稿怎么写
2014/08/26 职场文书
大学生自我评价200字(4篇)
2014/09/17 职场文书
2014年学校德育工作总结
2014/12/05 职场文书
研究生简历自我评
2015/03/11 职场文书
创卫工作总结2015
2015/04/22 职场文书
导游词之山海关
2019/12/10 职场文书
Java实现带图形界面的聊天程序
2022/06/10 Java/Android