浅谈多卡服务器下隐藏部分 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 相关文章推荐
Django添加favicon.ico图标的示例代码
Aug 07 Python
python使用wxpy轻松实现微信防撤回的方法
Feb 21 Python
Python基于opencv实现的简单画板功能示例
Mar 04 Python
python使用matplotlib绘制雷达图
Oct 18 Python
浅谈Python3实现两个矩形的交并比(IoU)
Jan 18 Python
django的模型类管理器——数据库操作的封装详解
Apr 01 Python
解决python父线程关闭后子线程不关闭问题
Apr 25 Python
MxNet预训练模型到Pytorch模型的转换方式
May 25 Python
python 常见的反爬虫策略
Sep 27 Python
Python Selenium XPath根据文本内容查找元素的方法
Dec 07 Python
python RSA加密的示例
Dec 09 Python
python套接字socket通信
Apr 01 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/06/23 PHP
thinkphp普通查询与表达式查询实例分析
2014/11/24 PHP
学习php设计模式 php实现享元模式(flyweight)
2015/12/07 PHP
php实现的后台表格分页功能示例
2017/10/23 PHP
JQuery 常用方法基础教程
2009/02/06 Javascript
使用jQuery重置(reset)表单的方法
2014/05/05 Javascript
js自定义鼠标右键的实现原理及源码
2014/06/23 Javascript
JavaScript DOM进阶方法
2015/04/13 Javascript
MVVM模式中ViewModel和View、Model有什么区别?
2015/06/19 Javascript
自定义Angular指令与jQuery实现的Bootstrap风格数据双向绑定的单选与多选下拉框
2015/12/12 Javascript
非常漂亮的相册集 使用jquery制作相册集
2016/04/28 Javascript
jQuery实现订单提交页发送短信功能前端处理方法
2016/07/04 Javascript
解析js如何获取css样式
2016/12/11 Javascript
微信小程序 数据封装,参数传值等经验分享
2017/01/09 Javascript
React Native如何消除启动时白屏的方法
2017/08/08 Javascript
解决option标签selected="selected"属性失效的问题
2017/11/06 Javascript
Node.js中console.log()输出彩色字体的方法示例
2019/12/01 Javascript
JS面向对象编程基础篇(三) 继承操作实例详解
2020/03/03 Javascript
[48:29]2018DOTA2亚洲邀请赛3月30日 小组赛A组 LGD VS KG
2018/03/31 DOTA
python实现定时播放mp3
2015/03/29 Python
分析并输出Python代码依赖的库的实现代码
2015/08/09 Python
python放大图片和画方格实现算法
2018/03/30 Python
python实现一行输入多个值和一行输出多个值的例子
2019/07/16 Python
ubuntu上安装python的实例方法
2019/09/30 Python
Series和DataFrame使用简单入门
2019/11/13 Python
python3实现从kafka获取数据,并解析为json格式,写入到mysql中
2019/12/23 Python
在pytorch 中计算精度、回归率、F1 score等指标的实例
2020/01/18 Python
tensorflow使用freeze_graph.py将ckpt转为pb文件的方法
2020/04/22 Python
彻底解决Python包下载慢问题
2020/11/15 Python
工业设计专业推荐信
2013/10/29 职场文书
小学生节约用水倡议书
2014/05/15 职场文书
韩语专业职业生涯规划范文:成功之路就在我们脚下
2014/09/11 职场文书
vue实现同时设置多个倒计时
2021/05/20 Vue.js
python字符串的多行输出的实例详解
2021/06/08 Python
python游戏开发之pygame实现接球小游戏
2022/04/22 Python
SQL Server使用PIVOT与unPIVOT实现行列转换
2022/05/25 SQL Server