运行tensorflow python程序,限制对GPU和CPU的占用操作


Posted in Python onFebruary 06, 2020

一般情况下,运行tensorflow时,默认会占用可以看见的所有GPU,那么就会导致其它用户或程序无GPU可用,那么就需要限制程序对GPU的占用。并且,一般我们的程序也用不了所有的GPU资源,只是强行霸占着,大部分资源都不会用到,也不会提升运行速度。

使用nvidia-smi可以查看本机的GPU使用情况,如下图,这里可以看出,本机的GPU型号是K80,共有两个K80,四块可用(一个K80包括两块K40)。

1、如果是只需要用某一块或某几块GPU,可以在运行程序时,利用如下命令运行:CUDA_VISIBLE_DEVICES=0,1 python test.py

这里表示只有GPU 0和1对程序可见,因此也就限制了程序只能用GPU 0和1

同样,也可以在代码里指定

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

如果想只用CPU,不用CPU来运行程序,可以用如下命令(所有GPU都不可见):

CUDA_VISIBLE_DEVICES='' python test.py

或者是

CUDA_VISIBLE_DEVICES="-1" python test.py

2、让tensorflow只按需索取显存,如下代码所示

#only minimum use gpu
gpu_config = tf.ConfigProto()
gpu_config.gpu_options.allow_growth = True
with tf.Session(config = gpu_config) as sess:

前面是对GPU的限制,那如果不用GPU,只用CPU呢?如何限制对CPU的使用呢?

前面也有提到,如果使用命令CUDA_VISIBLE_DEVICES=“”python test.py可以只使用CPU,那如果想只使用部分CPU呢?可以通过如下代码限制

cpu_config = tf.ConfigProto(intra_op_parallelism_threads = 8, inter_op_parallelism_threads = 8, device_count = {'CPU': 8})
with tf.Session(config = cpu_config) as sess:

以上这篇运行tensorflow python程序,限制对GPU和CPU的占用操作就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python difflib模块示例讲解
Sep 13 Python
python实现识别手写数字 python图像识别算法
Mar 23 Python
Python对CSV、Excel、txt、dat文件的处理
Sep 18 Python
python实现烟花小程序
Jan 30 Python
Python3最长回文子串算法示例
Mar 04 Python
python opencv调用笔记本摄像头
Aug 28 Python
python KNN算法实现鸢尾花数据集分类
Oct 24 Python
Pandas实现DataFrame按行求百分数(比例数)
Dec 27 Python
TensorFlow使用Graph的基本操作的实现
Apr 22 Python
解决pip install psycopg2出错问题
Jul 09 Python
Django封装交互接口代码
Jul 12 Python
python 列表推导和生成器表达式的使用
Feb 01 Python
如何在django中添加日志功能
Feb 06 #Python
keras tensorflow 实现在python下多进程运行
Feb 06 #Python
Scrapy框架实现的登录网站操作示例
Feb 06 #Python
Tensorflow 多线程设置方式
Feb 06 #Python
Scrapy框架基本命令与settings.py设置
Feb 06 #Python
python opencv圆、椭圆与任意多边形的绘制实例详解
Feb 06 #Python
Python输出指定字符串的方法
Feb 06 #Python
You might like
如何对PHP程序中的常见漏洞进行攻击(上)
2006/10/09 PHP
教你如何把一篇文章按要求分段
2006/10/09 PHP
用session做客户验证时的注意事项
2006/10/09 PHP
深入理解:单一入口、MVC、ORM、CURD、ActiveRecord概念
2013/06/06 PHP
Thinkphp实现短信验证注册功能
2016/10/18 PHP
PHP 二级子目录(后台目录)设置二级域名
2017/03/02 PHP
javascript获取当前ip的代码
2009/05/10 Javascript
JQuery为textarea添加maxlength属性的代码
2010/04/07 Javascript
你可能不知道的JavaScript的new Function()方法
2014/04/17 Javascript
自定义Angular指令与jQuery实现的Bootstrap风格数据双向绑定的单选与多选下拉框
2015/12/12 Javascript
2016年最热门的15 款代码语法高亮工具,美化你的代码
2016/01/06 Javascript
基于JavaScript实现快速转换文本语言(繁体中文和简体中文)
2016/03/07 Javascript
老生常谈JavaScript中的this关键字
2016/10/01 Javascript
完美的js图片轮换效果
2017/02/05 Javascript
JavaScript数据结构之二叉树的查找算法示例
2017/04/13 Javascript
elementUI Tree 树形控件的官方使用文档
2019/04/25 Javascript
vue自动路由-单页面项目(非build时构建)
2019/04/30 Javascript
JavaScript禁止右击保存图片,禁止拖拽图片的实现代码
2020/04/28 Javascript
vue+echarts+datav大屏数据展示及实现中国地图省市县下钻功能
2020/11/16 Javascript
[01:45]亚洲邀请赛互动指南虚拟物品介绍
2015/01/30 DOTA
[50:58]2018DOTA2亚洲邀请赛 4.1 小组赛 B组 Mineski vs EG
2018/04/03 DOTA
[56:12]LGD vs Optic Supermajor小组赛D组胜者组决赛 BO3 第一场 6.3
2018/06/04 DOTA
[01:36:19]Secret vs NB 2018国际邀请赛小组赛BO2 第一场 8.19
2018/08/21 DOTA
python抓取网页时字符集转换问题处理方案分享
2014/06/19 Python
python&MongoDB爬取图书馆借阅记录
2016/02/05 Python
Python变量赋值的秘密分享
2018/04/03 Python
numpy.transpose对三维数组的转置方法
2018/04/17 Python
对python:print打印时加u的含义详解
2018/12/15 Python
雪花秀美国官方网站:韩国著名草本护肤化妆品品牌
2016/10/19 全球购物
超市促销活动方案
2014/03/05 职场文书
公务员上班玩游戏检讨书
2014/09/17 职场文书
车辆委托书范本
2014/10/05 职场文书
公司保洁员岗位职责
2015/02/13 职场文书
小兵张嘎观后感300字
2015/06/03 职场文书
2016年小学植树节活动总结
2016/03/16 职场文书
《三国志》赏析
2019/08/27 职场文书