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 07 Python
python 中的int()函数怎么用
Oct 17 Python
Python实现统计给定字符串中重复模式最高子串功能示例
May 16 Python
使用Python进行QQ批量登录的实例代码
Jun 11 Python
python3 实现对图片进行局部切割的方法
Dec 05 Python
详解Python locals()的陷阱
Mar 26 Python
解决Django生产环境无法加载静态文件问题的解决
Apr 23 Python
django-allauth入门学习和使用详解
Jul 03 Python
Python如何获取Win7,Win10系统缩放大小
Jan 10 Python
Django如何使用jwt获取用户信息
Apr 21 Python
Django中ORM的基本使用教程
Dec 22 Python
python用分数表示矩阵的方法实例
Jan 11 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
How do I change MySQL timezone?
2008/03/26 PHP
非常实用的PHP常用函数汇总
2014/12/17 PHP
php 策略模式原理与应用深入理解
2019/09/25 PHP
newxtree.js代码
2007/03/13 Javascript
Javascript 模式实例 观察者模式
2009/10/24 Javascript
用jQuery打造TabPanel效果代码
2010/05/22 Javascript
Jquery多选框互相内容交换的实例代码
2013/07/04 Javascript
jQuery中andSelf()方法用法实例
2015/01/08 Javascript
js实现简单秒表走动的时钟特效
2020/03/25 Javascript
js图片轮播手动切换效果
2015/11/10 Javascript
jQuery点击其他地方时菜单消失的实现方法
2016/04/22 Javascript
基于Bootstrap实现的下拉菜单手机端不能选择菜单项的原因附解决办法
2016/07/22 Javascript
win系统下nodejs环境安装配置
2017/05/04 NodeJs
vue子组件使用自定义事件向父组件传递数据
2017/05/27 Javascript
详解webpack+angular2开发环境搭建
2017/06/28 Javascript
vue实现nav导航栏的方法
2017/12/13 Javascript
解决vue 更改计算属性后select选中值不更改的问题
2018/03/02 Javascript
vue+axios 前端实现登录拦截的两种方式(路由拦截、http拦截)
2018/10/24 Javascript
在微信小程序中渲染HTML内容3种解决方案及分析与问题解决
2020/01/12 Javascript
javascrpt密码强度校验函数详解
2020/03/18 Javascript
python 获取et和excel的版本号
2009/04/09 Python
python实现搜索本地文件信息写入文件的方法
2016/02/22 Python
python爬取百度贴吧前1000页内容(requests库面向对象思想实现)
2019/08/10 Python
tensorboard 可以显示graph,却不能显示scalar的解决方式
2020/02/15 Python
信号生成及DFT的python实现方式
2020/02/25 Python
使用Python实现将多表分批次从数据库导出到Excel
2020/05/15 Python
python 日志模块logging的使用场景及示例
2021/01/04 Python
CSS3新属性transition-property transform box-shadow实例学习
2013/06/06 HTML / CSS
结构工程研究生求职信
2013/10/13 职场文书
兼职学生的自我评价
2013/11/24 职场文书
乡镇庆八一活动方案
2014/02/02 职场文书
年度献血先进个人事迹材料
2014/02/14 职场文书
教学个人的自我评价分享
2014/02/16 职场文书
大学毕业生求职自荐书
2014/06/05 职场文书
2014年检验科工作总结
2014/11/22 职场文书
护士自我推荐信范文
2015/03/24 职场文书