浅谈多卡服务器下隐藏部分 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定时检查启动某个exe程序适合检测exe是否挂了
Jan 21 Python
pyqt和pyside开发图形化界面
Jan 22 Python
python中文乱码不着急,先看懂字节和字符
Dec 20 Python
Python中pandas模块DataFrame创建方法示例
Jun 20 Python
python对日志进行处理的实例代码
Oct 06 Python
利用arcgis的python读取要素的X,Y方法
Dec 22 Python
Python中如何导入类示例详解
Apr 17 Python
Python编写打字训练小程序
Sep 26 Python
Python中使用filter过滤列表的一个小技巧分享
May 02 Python
Python实现封装打包自己写的代码,被python import
Jul 12 Python
Python获取百度热搜的完整代码
Apr 07 Python
python和anaconda的区别
May 06 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
Protoss建筑一览
2020/03/14 星际争霸
PHP判断变量是否为0的方法
2014/02/08 PHP
Laravel中使用自己编写类库的3种方法
2015/02/10 PHP
PHP fopen函数用法实例讲解
2019/02/15 PHP
PHP队列场景以及实现代码实例详解
2021/02/26 PHP
CSS心形加载的动画源码的实现
2021/03/09 HTML / CSS
优化javascript的执行速度
2010/01/23 Javascript
javascript window.opener的用法分析
2010/04/07 Javascript
jquery中get和post的简单实例
2014/02/04 Javascript
JS实现CheckBox复选框全选全不选功能
2015/05/06 Javascript
纯javascript代码实现计算器功能(三种方法)
2015/09/07 Javascript
BootStrap智能表单实战系列(三)分块表单配置详解
2016/06/13 Javascript
原生js实现查询天气小应用
2016/12/09 Javascript
深入对Vue.js $watch方法的理解
2017/03/20 Javascript
Avalonjs双向数据绑定与监听的实例代码
2017/06/23 Javascript
Vue props用法详解(小结)
2018/07/03 Javascript
微信小程序实现一张或多张图片上传(云开发)
2019/09/25 Javascript
jQuery实现雪花飘落效果
2020/08/02 jQuery
NestJs使用Mongoose对MongoDB操作的方法
2021/02/22 Javascript
Opencv+Python 色彩通道拆分及合并的示例
2018/12/08 Python
django框架单表操作之增删改实例分析
2019/12/16 Python
Python集成开发工具Pycharm的安装和使用详解
2020/03/18 Python
selenium WebDriverWait类等待机制的实现
2020/03/18 Python
Python sql注入 过滤字符串的非法字符实例
2020/04/03 Python
解决Django no such table: django_session的问题
2020/04/07 Python
python 批量下载bilibili视频的gui程序
2020/11/20 Python
size?瑞典:英国伦敦的球鞋精品店
2018/03/01 全球购物
在Ajax应用中信息是如何在浏览器和服务器之间传递的
2016/05/31 面试题
简单的JAVA编程面试题
2013/03/19 面试题
企业环保标语
2014/06/10 职场文书
2015年党员承诺书
2015/01/21 职场文书
酒店人事专员岗位职责
2015/04/07 职场文书
超级礼物观后感
2015/06/15 职场文书
2015年小学语文教师工作总结
2015/10/23 职场文书
css 中多种边框的实现小窍门
2021/04/07 HTML / CSS
SpringBoot详解执行过程
2022/07/15 Java/Android