浅谈多卡服务器下隐藏部分 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输出一个杨辉三角的例子
Jun 13 Python
Python中实现字符串类型与字典类型相互转换的方法
Aug 18 Python
python字符串排序方法
Aug 29 Python
Python实现单词翻译功能
Jun 06 Python
Python md5与sha1加密算法用法分析
Jul 14 Python
Python批量更改文件名的实现方法
Oct 29 Python
python模仿网页版微信发送消息功能
Feb 24 Python
Python 生成 -1~1 之间的随机数矩阵方法
Aug 04 Python
python实现支付宝转账接口
May 07 Python
Django保护敏感信息的方法示例
May 09 Python
使用python实现对元素的长截图功能
Nov 14 Python
手把手教你安装Windows版本的Tensorflow
Mar 26 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对字符串的递增运算分析
2010/08/08 PHP
PHP定时执行计划任务的多种方法小结
2011/12/19 PHP
浅析php数据类型转换
2014/01/09 PHP
PHP连接MSSQL2008/2005数据库(SQLSRV)配置实例
2014/10/22 PHP
Laravel 5.2 文档 数据库 ―― 起步介绍
2019/10/21 PHP
页面中body onload 和 window.onload 冲突的问题的解决
2009/07/01 Javascript
jQuery验证Checkbox是否选中的代码 推荐
2011/09/04 Javascript
json2.js的初步学习与了解
2011/10/06 Javascript
jQuery层次选择器用法示例
2016/09/09 Javascript
Bootstarp基本模版学习教程
2017/02/01 Javascript
微信小程序之判断页面滚动方向的示例代码
2018/08/30 Javascript
Vue.js 十五分钟入门图文教程
2018/09/12 Javascript
JS设置自定义快捷键并实现图片上下左右移动
2019/10/17 Javascript
Vue2.0 $set()的正确使用详解
2020/07/28 Javascript
vue中路由跳转不计入history的操作
2020/09/21 Javascript
[03:24]CDEC.Y赛前采访 努力备战2016国际邀请赛中国区预选赛
2016/06/25 DOTA
Python实现删除Android工程中的冗余字符串
2015/01/19 Python
Python获取当前路径实现代码
2017/05/08 Python
python编写分类决策树的代码
2017/12/21 Python
python实现外卖信息管理系统
2018/01/11 Python
python实现对服务器脚本敏感信息的加密解密功能
2019/08/13 Python
解决Pycharm中恢复被exclude的项目问题(pycharm source root)
2020/02/14 Python
python 解压、复制、删除 文件的实例代码
2020/02/26 Python
雅诗兰黛美国官网:Estee Lauder美国
2016/07/21 全球购物
如果让你测试一台高速激光打印机,你都会进行哪些测试
2012/12/04 面试题
工程预算与管理应届生求职信
2013/10/06 职场文书
高中生自我评价个人范文
2013/11/09 职场文书
施工资料员岗位职责
2014/01/06 职场文书
幼儿园招生广告
2014/03/19 职场文书
校园绿化美化方案
2014/06/08 职场文书
逃出克隆岛观后感
2015/06/09 职场文书
北京青年观后感
2015/06/15 职场文书
2016年教师节感恩寄语
2015/12/04 职场文书
高中数学课堂教学反思
2016/02/18 职场文书
CSS布局之浮动(float)和定位(position)属性的区别
2021/09/25 HTML / CSS
Spring boot admin 服务监控利器详解
2022/08/05 Java/Android