运行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 相关文章推荐
win7上python2.7连接mysql数据库的方法
Jan 14 Python
快速入门python学习笔记
Dec 06 Python
TensorFlow中权重的随机初始化的方法
Feb 11 Python
用Django写天气预报查询网站
Oct 21 Python
在Python中表示一个对象的方法
Jun 25 Python
详解Python并发编程之从性能角度来初探并发编程
Aug 23 Python
tesserocr与pytesseract模块的使用方法解析
Aug 30 Python
Python爬取爱奇艺电影信息代码实例
Nov 26 Python
tensorflow 实现数据类型转换
Feb 17 Python
python实现简单的tcp 文件下载
Sep 16 Python
Python实现天气查询软件
Jun 07 Python
什么是Python装饰器?如何定义和使用?
Apr 11 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 cookis创建实现代码
2009/03/16 PHP
PHP加密解密函数详解
2015/10/28 PHP
thinkPHP简单导入和使用阿里云OSSsdk的方法
2017/03/15 PHP
javascript中attribute和property的区别详解
2014/06/05 Javascript
JavaScript实现检查页面上的广告是否被AdBlock屏蔽了的方法
2014/11/03 Javascript
提升PHP安全:8个必须修改的PHP默认配置
2014/11/17 Javascript
javascript表单验证和Window详解
2014/12/11 Javascript
jQuery UI插件自定义confirm确认框的方法
2015/03/20 Javascript
JavaScript 七大技巧(一)
2015/12/13 Javascript
在WordPress中加入Google搜索功能的简单步骤讲解
2016/01/04 Javascript
JS三级可折叠菜单实现方法
2016/02/29 Javascript
纯js实现手风琴效果
2020/04/17 Javascript
JS中如何实现复选框全选功能
2016/12/19 Javascript
EasyUI学习之Combobox下拉列表(1)
2016/12/29 Javascript
node.js操作mysql简单实例
2017/05/25 Javascript
如何获取TypeScript的声明文件.d.ts
2018/05/01 Javascript
轻量级富文本编辑器wangEditor结合vue使用方法示例
2018/10/10 Javascript
基于nodejs的雪碧图制作工具的示例代码
2018/11/05 NodeJs
使用webpack和rollup打包组件库的方法
2021/02/25 Javascript
[04:48]DOTA2上海特锦赛小组赛第三日 TOP10精彩集锦
2016/02/28 DOTA
[02:17]2016国际邀请赛中国区预选赛VG战队领队采访
2016/06/26 DOTA
对python 匹配字符串开头和结尾的方法详解
2018/10/27 Python
python创造虚拟环境方法总结
2019/03/04 Python
Python实现决策树并且使用Graphviz可视化的例子
2019/08/09 Python
Django框架序列化与反序列化操作详解
2019/11/01 Python
Python调用接口合并Excel表代码实例
2020/03/31 Python
详解Python中如何将数据存储为json格式的文件
2020/11/18 Python
python 实现socket服务端并发的四种方式
2020/12/14 Python
世界最大的私人旅行指南出版商:孤独星球
2016/08/23 全球购物
档案检查欢迎词
2014/01/13 职场文书
社区道德讲堂实施方案
2014/03/21 职场文书
销售岗位职责范本
2014/06/12 职场文书
综治工作汇报材料
2014/10/27 职场文书
水浒传读书笔记
2015/06/25 职场文书
详解Python小数据池和代码块缓存机制
2021/04/07 Python
Nginx本地配置SSL访问的实例教程
2022/05/30 Servers