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中使用__slots__方法的详细教程
Apr 28 Python
探究Python的Tornado框架对子域名和泛域名的支持
May 02 Python
python利用OpenCV2实现人脸检测
Apr 16 Python
python+opencv实现动态物体识别
Jan 09 Python
简单实现Python爬取网络图片
Apr 01 Python
Python Selenium 之关闭窗口close与quit的方法
Feb 13 Python
pytorch 可视化feature map的示例代码
Aug 20 Python
Python类中的魔法方法之 __slots__原理解析
Aug 26 Python
Python3 main函数使用sys.argv传入多个参数的实现
Dec 25 Python
TensorFlow tf.nn.conv2d实现卷积的方式
Jan 03 Python
python使用openpyxl操作excel的方法步骤
May 28 Python
用python写PDF转换器的实现
Oct 29 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入门小知识
2008/03/24 PHP
PHP管理内存函数 memory_get_usage()使用介绍
2012/09/23 PHP
PHP如何使用Memcached
2016/04/05 PHP
PHP 中TP5 Request 请求对象的实例详解
2017/07/31 PHP
php无限级分类实现评论及回复功能
2019/02/18 PHP
javascript校验价格合法性实例(必须输入2位小数)
2014/05/05 Javascript
js实现的点击数量加一可操作数据库
2014/05/09 Javascript
javascript实现数独解法
2015/03/14 Javascript
新入门node.js必须要知道的概念(必看篇)
2016/08/10 Javascript
基于JS设计12306登录页面
2016/12/28 Javascript
Vue自定义指令使用方法详解
2017/08/21 Javascript
[js高手之路]图解javascript的原型(prototype)对象,原型链实例
2017/08/28 Javascript
javascript流程控制语句集合
2017/09/18 Javascript
Three.js实现浏览器变动时进行自适应的方法
2017/09/26 Javascript
vue父组件向子组件(props)传递数据的方法
2018/01/02 Javascript
vue打包之后生成一个配置文件修改接口的方法
2018/12/09 Javascript
如何为你的JS项目添加智能提示与类型检查详解
2019/03/12 Javascript
详解JavaScript中的坐标和距离
2019/05/27 Javascript
node.js开发辅助工具nodemon安装与配置详解
2020/02/06 Javascript
Python的Django框架中的select_related函数对QuerySet 查询的优化
2015/04/01 Python
Python远程桌面协议RDPY安装使用介绍
2015/04/15 Python
如何安装多版本python python2和python3共存以及pip共存
2018/09/18 Python
Django unittest 设置跳过某些case的方法
2018/12/26 Python
python 实现提取某个索引中某个时间段的数据方法
2019/02/01 Python
PyQt5多线程刷新界面防假死示例
2019/12/13 Python
pandas中ix的使用详细讲解
2020/03/09 Python
使用Python和百度语音识别生成视频字幕的实现
2020/04/09 Python
Python中常见的数制转换有哪些
2020/05/27 Python
CSS3弹性盒模型开发笔记(二)
2016/04/26 HTML / CSS
HTML5标签与HTML4标签的区别示例介绍
2013/07/18 HTML / CSS
美国浴缸、水槽和水龙头购物网站:Vintage Tub & Bath
2019/11/05 全球购物
几个Linux面试题笔试题
2012/12/01 面试题
环保专项行动方案
2014/05/12 职场文书
财务人员个人工作总结
2015/02/27 职场文书
八年级物理教学反思
2016/02/19 职场文书
预备党员的思想汇报,你真的会写吗?
2019/06/28 职场文书