浅谈多卡服务器下隐藏部分 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实现PS图像调整黑白效果示例
Jan 25 Python
Python实现定时精度可调节的定时器
Apr 15 Python
Python对多属性的重复数据去重实例
Apr 18 Python
Python 3.8中实现functools.cached_property功能
May 29 Python
python实现DEM数据的阴影生成的方法
Jul 23 Python
Python使用Pandas库常见操作详解
Jan 16 Python
python爬虫把url链接编码成gbk2312格式过程解析
Jun 08 Python
pytorch随机采样操作SubsetRandomSampler()
Jul 07 Python
python绘制趋势图的示例
Sep 17 Python
python如何利用paramiko执行服务器命令
Nov 07 Python
Python 实现PS滤镜中的径向模糊特效
Dec 03 Python
对PyTorch中inplace字段的全面理解
May 22 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
php设计模式小结
2013/02/15 PHP
WordPress的文章自动添加关键词及关键词的SEO优化
2016/03/01 PHP
php、mysql查询当天,查询本周,查询本月的数据实例(字段是时间戳)
2017/02/04 PHP
ThinkPHP实现图片上传操作的方法详解
2017/05/08 PHP
定义select的边框颜色
2008/04/28 Javascript
Javascript学习笔记1 数据类型
2010/01/11 Javascript
Textbox控件注册回车事件及触发按钮提交事件具体实现
2013/03/04 Javascript
js兼容pc端浏览器并有多种弹出小提示的手机端浮层控件实例
2015/04/29 Javascript
如何防止JavaScript自动插入分号
2015/11/05 Javascript
Bootstrap安装环境配置教程分享
2016/05/27 Javascript
JavaScript日期对象(Date)基本用法示例
2017/01/18 Javascript
thinkjs 文件上传功能实例代码
2017/11/08 Javascript
利用vue组件自定义v-model实现一个Tab组件方法示例
2017/12/06 Javascript
使用Vue中 v-for循环列表控制按钮隐藏显示功能
2019/04/23 Javascript
vue实现前台列表数据过滤搜索、分页效果
2019/05/28 Javascript
Vue axios 将传递的json数据转为form data的例子
2019/10/29 Javascript
Vue中常用rules校验规则(实例代码)
2019/11/14 Javascript
[04:04]显微镜下的DOTA2第六期——电影级别的华丽团战
2014/06/20 DOTA
python实现带错误处理功能的远程文件读取方法
2015/04/29 Python
wx.CheckBox创建复选框控件并响应鼠标点击事件
2018/04/25 Python
python爬虫之验证码篇3-滑动验证码识别技术
2019/04/11 Python
python定间隔取点(np.linspace)的实现
2019/11/27 Python
解决Python二维数组赋值问题
2019/11/28 Python
详解python 破解网站反爬虫的两种简单方法
2020/02/09 Python
python 对象真假值的实例(哪些视为False)
2020/12/11 Python
南威尔士家居商店:Leekes
2016/10/25 全球购物
理工大学毕业生自荐信范文
2014/02/22 职场文书
保密工作责任书
2014/04/16 职场文书
家长建议怎么写
2014/05/15 职场文书
环境卫生标语
2014/06/09 职场文书
酒店管理失职检讨书
2014/09/16 职场文书
社区服务活动报告
2015/02/05 职场文书
复活读书笔记
2015/06/29 职场文书
大学生先进个人主要事迹材料
2015/11/04 职场文书
2019年七夕情人节浪漫祝福语大全!
2019/08/08 职场文书
python自动化测试之Selenium详解
2022/03/13 Python