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+Django在windows下的开发环境配置图解
Nov 11 Python
python 切片和range()用法说明
Mar 24 Python
python字典快速保存于读取的方法
Mar 23 Python
Python 实现引用其他.py文件中的类和类的方法
Apr 29 Python
python版DDOS攻击脚本
Jun 12 Python
python交易记录整合交易类详解
Jul 03 Python
Python中Subprocess的不同函数解析
Dec 10 Python
tensorflow 利用expand_dims和squeeze扩展和压缩tensor维度方式
Feb 07 Python
Pytorch mask_select 函数的用法详解
Feb 18 Python
django queryset相加和筛选教程
May 18 Python
Django 构建模板form表单的两种方法
Jun 14 Python
基于Python实现对比Exce的工具
Apr 07 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
如何提高MYSQL数据库的查询统计速度 select 索引应用
2007/04/11 PHP
一个无限级XML绑定跨框架菜单(For IE)
2007/01/27 Javascript
OfflineSave离线保存代码再次发布使用说明
2007/05/23 Javascript
javascript Xml增删改查(IE下)操作实现代码
2009/01/30 Javascript
jQuery LigerUI 插件介绍及使用之ligerDrag和ligerResizable示例代码打包
2011/04/06 Javascript
jQuery 获取URL的GET参数值的小例子
2013/04/18 Javascript
Javascript中3种实现继承的方法和代码实例
2014/08/12 Javascript
浏览器缩放检测的js代码
2014/09/28 Javascript
jQuery实现按钮只点击一次后就取消点击事件绑定的方法
2015/06/26 Javascript
nodejs加密Crypto的实例代码
2016/07/07 NodeJs
AngularJs ng-repeat 嵌套如何获取外层$index
2016/09/21 Javascript
Redux实现组合计数器的示例代码
2018/07/04 Javascript
浅谈微信小程序flex布局基础
2018/09/10 Javascript
从零搭一个自用的前端脚手架的方法步骤
2019/09/23 Javascript
如何基于layui的laytpl实现数据绑定的示例代码
2020/04/10 Javascript
Python返回真假值(True or False)小技巧
2015/04/10 Python
Flask 让jsonify返回的json串支持中文显示的方法
2018/03/26 Python
python批量赋值操作实例
2018/10/22 Python
Django 接收Post请求数据,并保存到数据库的实现方法
2019/07/12 Python
python opencv将图片转为灰度图的方法示例
2019/07/31 Python
Python用5行代码实现批量抠图的示例代码
2020/04/14 Python
浅谈Python协程
2020/06/17 Python
如何在python中判断变量的类型
2020/07/29 Python
Python自动化之UnitTest框架实战记录
2020/09/08 Python
CSS Grid布局教程之浏览器开启CSS Grid Layout汇总
2014/12/30 HTML / CSS
印度和世界各地的精美产品:Ikka Dukka
2018/02/12 全球购物
如何开启linux的ssh服务
2015/02/14 面试题
大学生个人简历自我评价
2013/11/16 职场文书
应届大学生自荐信
2013/12/05 职场文书
高考标语大全
2014/06/05 职场文书
农民工预备党员思想汇报
2014/09/14 职场文书
2014年防汛工作总结
2014/12/08 职场文书
2015年街道办事处工作总结
2015/05/22 职场文书
学校少先队工作总结
2015/08/12 职场文书
Nginx服务器添加Systemd自定义服务过程解析
2021/03/31 Servers
一文解答什么是MySQL的回表
2022/08/05 MySQL