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类的用法实例浅析
May 27 Python
python实现上传下载文件功能
Nov 19 Python
python数据结构之链表详解
Sep 12 Python
浅谈tensorflow1.0 池化层(pooling)和全连接层(dense)
Apr 27 Python
Python面向对象之类的内置attr属性示例
Dec 14 Python
Python 做曲线拟合和求积分的方法
Dec 29 Python
python学生管理系统学习笔记
Mar 19 Python
在Python中如何传递任意数量的实参的示例代码
Mar 21 Python
详细整理python 字符串(str)与列表(list)以及数组(array)之间的转换方法
Aug 30 Python
python通过SSH登陆linux并操作的实现
Oct 10 Python
Python批量删除mysql中千万级大量数据的脚本分享
Dec 03 Python
超级详细实用的pycharm常用快捷键
May 12 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简单实现加减乘除计算器
2014/01/06 PHP
PHP使用redis实现统计缓存mysql压力的方法
2015/11/14 PHP
javascript获得CheckBoxList选中的数量
2009/10/27 Javascript
理解Javascript_01_理解内存分配原理分析
2010/10/11 Javascript
什么是DOM(Document Object Model)文档对象模型
2012/03/05 Javascript
jQuery判断iframe中元素是否存在的方法
2013/05/11 Javascript
jQuery实现点击文本框弹出热门标签的提示效果
2013/11/17 Javascript
Chrome下ifame父窗口调用子窗口的问题示例探讨
2014/03/17 Javascript
原生js获取宽高与jquery获取宽高的方法关系对比
2014/04/04 Javascript
jQuery中val()方法用法实例
2014/12/25 Javascript
jQuery中end()方法用法实例
2015/01/08 Javascript
Javascript 是你的高阶函数(高级应用)
2015/06/15 Javascript
JS组件Bootstrap Table布局详解
2016/05/27 Javascript
AngularJS框架的ng-app指令与自动加载实现方法分析
2017/01/04 Javascript
深入理解Javascript中的valueOf与toString
2017/01/04 Javascript
详谈Ajax请求中的async:false/true的作用(ajax 在外部调用问题)
2017/02/10 Javascript
在vue中使用SockJS实现webSocket通信的过程
2018/08/29 Javascript
详解vuex状态管理模式
2018/11/01 Javascript
JS实现处理时间,年月日,星期的公共方法示例
2019/05/31 Javascript
解决vue中使用proxy配置不同端口和ip接口问题
2019/08/14 Javascript
微信小程序之下拉列表实现方法解析(附完整源码)
2019/08/23 Javascript
vue实现配置全局访问路径头(axios)
2019/11/01 Javascript
浅谈如何优雅处理JavaScript异步错误
2019/11/12 Javascript
vue-cli3.0实现一个多页面应用的历奇经历记录总结
2020/03/16 Javascript
python字符串编码识别模块chardet简单应用
2015/06/15 Python
python单例模式实例解析
2018/08/28 Python
Python实现的删除重复文件或图片功能示例【去重】
2019/04/23 Python
python 比较2张图片的相似度的方法示例
2019/12/18 Python
Python 项目转化为so文件实例
2019/12/23 Python
python 删除系统中的文件(按时间,大小,扩展名)
2020/11/19 Python
村主任“四风”问题个人整改措施
2014/10/04 职场文书
继承权公证书范本
2015/01/23 职场文书
2016年社会主义核心价值观心得体会
2016/01/21 职场文书
python 逐步回归算法
2021/04/06 Python
教你怎么用Python selenium操作浏览器对象的基础API
2021/06/23 Python
Windows11性能真的上涨35%? 桌面酷睿i9实测结果公开
2021/11/21 数码科技