浅谈多卡服务器下隐藏部分 GPU 和 TensorFlow 的显存使用设置


Posted in Python onJune 30, 2020

服务器有多张显卡,一般是组里共用,分配好显卡和任务就体现公德了。除了在代码中指定使用的 GPU 编号,还可以直接设置可见 GPU 编号,使程序/用户只对部分 GPU 可见。

操作很简单,使用环境变量 CUDA_VISIBLE_DEVICES 即可。

具体来说,如果使用单卡运行 Python 脚本,则可输入

CUDA_VISIBLE_DEVICES=1 python my_script.py

脚本将只使用 GPU1。

在 .py 脚本和 Notebook 中设置,则

import os

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

还可以直接设置临时的环境变量:

export CUDA_VISIBLE_DEVICES="0"

此时该用户的 CUDA 只看得见 GPU0。

至于显存设置,可以设置使用比例(70%):

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

也可以按需增长:

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)

如果是 Keras 使用 TensorFlow 后端,则可通过如

import tensorflow as tf 
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
set_session(tf.Session(config=config))

更改使用设置。

以上这篇浅谈多卡服务器下隐藏部分 GPU 和 TensorFlow 的显存使用设置就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
剖析Python的Tornado框架中session支持的实现代码
Aug 21 Python
详解python发送各类邮件的主要方法
Dec 22 Python
Python+pandas计算数据相关系数的实例
Jul 03 Python
Python Excel处理库openpyxl使用详解
May 09 Python
Python  Django 母版和继承解析
Aug 09 Python
django多种支付、并发订单处理实例代码
Dec 13 Python
利用PyCharm操作Github(仓库新建、更新,代码回滚)
Dec 18 Python
Django模板之基本的 for 循环 和 List内容的显示方式
Mar 31 Python
关于keras.layers.Conv1D的kernel_size参数使用介绍
May 22 Python
Python‘==‘ 及 ‘is‘相关原理解析
Sep 05 Python
Python+OpenCV检测灯光亮点的实现方法
Nov 02 Python
分享Python获取本机IP地址的几种方法
Mar 17 Python
Tensorflow中批量读取数据的案列分析及TFRecord文件的打包与读取
Jun 30 #Python
使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)
Jun 30 #Python
keras的backend 设置 tensorflow,theano操作
Jun 30 #Python
浅谈TensorFlow中读取图像数据的三种方式
Jun 30 #Python
python中 _、__、__xx__()区别及使用场景
Jun 30 #Python
Django实现内容缓存实例方法
Jun 30 #Python
Pytorch 卷积中的 Input Shape用法
Jun 29 #Python
You might like
解决phpmyadmin 乱码,支持gb2312和utf-8
2006/11/20 PHP
PHP输出缓冲控制Output Control系列函数详解
2015/07/02 PHP
PHP的自定义模板引擎
2017/03/24 PHP
Laravel框架中队列和工作(Queues、Jobs)操作实例详解
2020/04/06 PHP
用JQuery调用Session的实现代码
2010/10/29 Javascript
JavaScript高级程序设计 读书笔记之十 本地对象Date日期
2012/02/27 Javascript
jQuery基础框架浅入剖析
2012/12/27 Javascript
怎么选择Javascript框架(Javascript Framework)
2013/11/22 Javascript
基于JQuery实现的图片自动进行缩放和裁剪处理
2014/01/31 Javascript
jQuery提示效果代码分享
2014/11/20 Javascript
js禁止浏览器页面后退功能的实例(推荐)
2017/09/01 Javascript
微信小程序实现点击按钮移动view标签的位置功能示例【附demo源码下载】
2017/12/06 Javascript
vue实现路由切换改变title功能
2019/05/28 Javascript
layer更改皮肤的实现方法
2019/09/11 Javascript
python计算对角线有理函数插值的方法
2015/05/07 Python
Python执行时间的计算方法小结
2017/03/17 Python
对python sklearn one-hot编码详解
2018/07/10 Python
Python实现处理逆波兰表达式示例
2018/07/30 Python
numpy中loadtxt 的用法详解
2018/08/03 Python
解决python ogr shp字段写入中文乱码的问题
2018/12/31 Python
python实现简单聊天室功能 可以私聊
2019/07/12 Python
Jupyter 无法下载文件夹如何实现曲线救国
2020/04/22 Python
Python实现封装打包自己写的代码,被python import
2020/07/12 Python
Python下使用Trackbar实现绘图板
2020/10/27 Python
Prometheus开发中间件Exporter过程详解
2020/11/30 Python
Python爬虫+tkinter界面实现历史天气查询的思路详解
2021/02/22 Python
全球地下的服装和态度:Slam Jam
2018/02/04 全球购物
会计岗位职责模板
2014/03/12 职场文书
药店促销活动策划方案
2014/08/24 职场文书
党的群众路线教育实践活动个人整改方案
2014/09/21 职场文书
2014年学前班工作总结
2014/12/08 职场文书
九九重阳节致辞
2015/07/31 职场文书
药房管理制度范本
2015/08/06 职场文书
2015年店长个人工作总结
2015/10/23 职场文书
导游词之临安白水涧
2019/11/05 职场文书
DIY胆机必读:各国电子管评价
2022/04/06 无线电