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进程的信号通讯
Apr 09 Python
python使用smtplib模块通过gmail实现邮件发送的方法
May 08 Python
在Python中操作字符串之rstrip()方法的使用
May 19 Python
python自带的http模块详解
Nov 06 Python
Python语言描述随机梯度下降法
Jan 04 Python
Python os.access()用法实例
Feb 18 Python
Appium+python自动化之连接模拟器并启动淘宝APP(超详解)
Jun 17 Python
Python 操作 ElasticSearch的完整代码
Aug 04 Python
python selenium xpath定位操作
Sep 01 Python
Python中的None与 NULL(即空字符)的区别详解
Sep 24 Python
用python画城市轮播地图
May 28 Python
如何在pycharm中快捷安装pip命令(如pygame)
May 31 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
论坛头像随机变换代码
2006/10/09 PHP
用PHP 4.2书写安全的脚本
2006/10/09 PHP
UTF8编码内的繁简转换的PHP类
2009/07/09 PHP
thinkphp微信开之安全模式消息加密解密不成功的解决办法
2015/12/02 PHP
php判断是否连接上网络的方法实例详解
2016/12/14 PHP
Laravel框架实现调用百度翻译API功能示例
2019/05/30 PHP
JavaScript几种形式的树结构菜单
2010/05/10 Javascript
jQuery :nth-child前有无空格的区别分析
2011/07/11 Javascript
ASP.NET jQuery 实例6 (实现CheckBoxList成员全选或全取消)
2012/01/13 Javascript
Js+Flash实现访问剪切板操作
2012/11/20 Javascript
js获取单选框或复选框值及操作
2012/12/18 Javascript
javascript里模拟sleep(两种实现方式)
2013/01/25 Javascript
解析javascript 浏览器关闭事件
2013/07/08 Javascript
ExtJS4 动态生成的grid导出为excel示例
2014/05/02 Javascript
深入分析Cookie的安全性问题
2015/03/01 Javascript
JavaScript运动减速效果实例分析
2015/08/04 Javascript
基于jQuery实现多层次的手风琴效果附源码
2015/09/21 Javascript
js和C# 时间日期格式转换的简单实例
2016/05/28 Javascript
javascript遍历json对象的key和任意js对象属性实例
2017/03/09 Javascript
js 将canvas生成图片保存,或直接保存一张图片的实现方法
2018/01/02 Javascript
js 取消页面可以选中文字的功能方法
2018/01/02 Javascript
[03:36]2015国际邀请赛第二日现场精彩集锦
2015/08/06 DOTA
[42:20]Winstrike vs VGJ.S 2018国际邀请赛淘汰赛BO3 第二场 8.23
2018/08/24 DOTA
Python实现模拟登录网易邮箱的方法示例
2018/07/05 Python
keras实现VGG16方式(预测一张图片)
2020/07/07 Python
美国最佳选择产品网站:Best Choice Products
2019/05/27 全球购物
linux面试题参考答案(8)
2015/08/11 面试题
日语专业个人的求职信
2013/12/03 职场文书
编辑求职信样本
2013/12/16 职场文书
开业庆典主持词
2014/03/21 职场文书
保护环境建议书300字
2014/05/13 职场文书
农村党支部书记司法四风问题对照检查材料
2014/09/26 职场文书
代办社保委托书范文
2014/10/06 职场文书
党风廉正建设个人工作总结
2015/03/06 职场文书
正规欠条模板
2015/07/03 职场文书
详解Python如何批量采集京东商品数据流程
2022/01/22 Python