tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现


Posted in Python onFebruary 06, 2020

在Python代码中指定GPU

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

设置定量的GPU使用量:

config = tf.ConfigProto()

config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存

session = tf.Session(config=config)

设置最小的GPU使用量:

config = tf.ConfigProto()

config.gpu_options.allow_growth = True

session = tf.Session(config=config)

以上这篇tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中optparse模块使用浅析
Jan 01 Python
Python数据类型学习笔记
Jan 13 Python
利用Python实现Windows定时关机功能
Mar 21 Python
使用Python通过win32 COM实现Word文档的写入与保存方法
May 08 Python
对Python中9种生成新对象的方法总结
May 23 Python
解决pycharm py文件运行后停止按钮变成了灰色的问题
Nov 29 Python
在Python中表示一个对象的方法
Jun 25 Python
NumPy中的维度Axis详解
Nov 26 Python
Python 词典(Dict) 加载与保存示例
Dec 06 Python
Python 装饰器原理、定义与用法详解
Dec 07 Python
python 多维高斯分布数据生成方式
Dec 09 Python
python获取引用对象的个数方式
Dec 20 Python
Tensorflow中tf.ConfigProto()的用法详解
Feb 06 #Python
Python continue语句实例用法
Feb 06 #Python
Tensorflow限制CPU个数实例
Feb 06 #Python
如何解决tensorflow恢复模型的特定值时出错
Feb 06 #Python
python range实例用法分享
Feb 06 #Python
tensorflow 报错unitialized value的解决方法
Feb 06 #Python
TensorFlow实现保存训练模型为pd文件并恢复
Feb 06 #Python
You might like
PHP print类函数使用总结
2010/06/25 PHP
PHP上传图片类显示缩略图功能
2016/06/30 PHP
Yii 2.0中场景的使用教程
2017/06/02 PHP
PHP 中TP5 Request 请求对象的实例详解
2017/07/31 PHP
PHP INT类型在内存中占字节详解
2019/07/20 PHP
{}与function(){}选用空对象{}来存放keyValue
2012/05/23 Javascript
js弹出确认是否删除对话框
2014/03/27 Javascript
Javascript中Date类型和Math类型详解
2016/02/27 Javascript
微信小程序 教程之注册页面
2016/10/17 Javascript
详解Vue.js基于$.ajax获取数据并与组件的data绑定
2017/05/26 Javascript
angular.js和vue.js中实现函数去抖示例(debounce)
2018/01/18 Javascript
小程序实现人脸识别功能(百度ai)
2018/12/23 Javascript
jquery获取img的src值实例介绍
2019/01/16 jQuery
微信小程序 bindtap 传参的实例代码
2020/02/21 Javascript
vue 子组件watch监听不到prop的解决
2020/08/09 Javascript
echarts柱状图背景重叠组合而非并列的实现代码
2020/12/10 Javascript
JavaScript实现切换多张图片
2021/01/27 Javascript
Python抓取电影天堂电影信息的代码
2016/04/07 Python
PyQt5实现下载进度条效果
2018/04/19 Python
详解Django中类视图使用装饰器的方式
2018/08/12 Python
Python基础学习之函数方法实例详解
2019/06/18 Python
Python中的引用和拷贝实例解析
2019/11/14 Python
python 实现矩阵按对角线打印
2019/11/29 Python
Python爬虫爬取煎蛋网图片代码实例
2019/12/16 Python
Python3 使用selenium插件爬取苏宁商家联系电话
2019/12/23 Python
python 装饰器的使用示例
2020/10/10 Python
Python爬虫实战案例之爬取喜马拉雅音频数据详解
2020/12/07 Python
python 录制系统声音的示例
2020/12/21 Python
挪威手表购物网站:Klokker
2016/09/19 全球购物
城野医生官方海外旗舰店:风靡亚洲毛孔收敛水
2018/04/26 全球购物
Abbott Lyon官网:女士手表、珠宝及配件
2020/12/26 全球购物
2014派出所所长群众路线对照检查材料思想汇报
2014/09/18 职场文书
出国留学导师推荐信
2015/03/26 职场文书
中学社团活动总结
2015/05/07 职场文书
2016年暑假家长对孩子评语
2015/12/01 职场文书
缓存替换策略及应用(以Redis、InnoDB为例)
2021/07/25 Redis