keras 指定程序在某块卡上训练实例


Posted in Python onJune 22, 2020

场景:某台机器上有三块卡,想同时开三个程序,放到三块卡上去训练。

策略:CUDA_VISIBLE_DEVICES=1 python train.py就可以指定程序在某块卡上训练。

keras 指定程序在某块卡上训练实例

补充知识:keras指定GPU及显存使用量

指定GPU

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

指定GPU和显存使用量

import os
from keras.backend.tensorflow_backend import set_session

os.environ["CUDA_VISIBLE_DEVICES"] = "0"
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
set_session(tf.Session(config=config))

指定GPU显存使用按需分配

import keras.backend.tensorflow_backend as KTF
import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0"
config = tf.ConfigProto()
config.gpu_options.allow_growth=True 
sess = tf.Session(config=config)
KTF.set_session(sess)

以上这篇keras 指定程序在某块卡上训练实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python语言编写电脑时间自动同步小工具
Mar 08 Python
python用字典统计单词或汉字词个数示例
Apr 22 Python
基于windows下pip安装python模块时报错总结
Jun 12 Python
使用CodeMirror实现Python3在线编辑器的示例代码
Jan 14 Python
Python日志无延迟实时写入的示例
Jul 11 Python
详解pandas中MultiIndex和对象实际索引不一致问题
Jul 23 Python
Python爬虫:将headers请求头字符串转为字典的方法
Aug 21 Python
python实现人工智能Ai抠图功能
Sep 05 Python
python监控nginx端口和进程状态
Sep 06 Python
Python对Excel按列值筛选并拆分表格到多个文件的代码
Nov 05 Python
利用Python制作动态排名图的实现代码
Apr 09 Python
Python装饰器详细介绍
Mar 25 Python
python Socket网络编程实现C/S模式和P2P
Jun 22 #Python
Python手动或自动协程操作方法解析
Jun 22 #Python
keras使用Sequence类调用大规模数据集进行训练的实现
Jun 22 #Python
Python socket服务常用操作代码实例
Jun 22 #Python
Python如何实现后端自定义认证并实现多条件登陆
Jun 22 #Python
零基础小白多久能学会python
Jun 22 #Python
Keras-多输入多输出实例(多任务)
Jun 22 #Python
You might like
模仿OSO的论坛(三)
2006/10/09 PHP
谈谈新手如何学习PHP
2006/12/23 PHP
php项目打包方法
2008/02/18 PHP
php中二维数组排序问题方法详解
2015/08/28 PHP
PHP函数rtrim()使用中的怪异现象分析
2017/02/24 PHP
解析JavaScript中的标签语句
2013/06/19 Javascript
js中cookie的添加、取值、删除示例代码
2013/10/21 Javascript
JavaScript检查某个function是否是原生代码的方法
2014/08/20 Javascript
JavaScript实现的内存数据库LokiJS介绍和入门实例
2014/11/17 Javascript
js实现带按钮的上下滚动效果
2015/05/12 Javascript
JavaScript中的getDay()方法使用详解
2015/06/09 Javascript
jQuery实现鼠标滑过链接控制图片的滑动展开与隐藏效果
2015/10/28 Javascript
JavaScript实现多种排序算法
2016/02/24 Javascript
Js获取当前日期时间及格式化代码
2016/09/17 Javascript
jQuery中的siblings()是什么意思(推荐)
2016/12/29 Javascript
vue中使用cropperjs的方法
2018/03/01 Javascript
详解react、redux、react-redux之间的关系
2018/04/11 Javascript
详解为element-ui的Select和Cascader添加弹层底部操作按钮
2020/02/07 Javascript
js实现跳一跳小游戏
2020/07/31 Javascript
IDEA配置jQuery, $符号不再显示黄色波浪线的问题
2020/10/09 jQuery
[01:05:56]Liquid vs VP Supermajor决赛 BO 第二场 6.10
2018/07/04 DOTA
[51:29]完美世界DOTA2联赛循环赛 Matador vs Forest BO2第一场 11.05
2020/11/05 DOTA
Python基于identicon库创建类似Github上用的头像功能
2017/09/25 Python
利用python实现简单的邮件发送客户端示例
2017/12/23 Python
在python中利用opencv简单做图片比对的方法
2019/01/24 Python
Python分析彩票记录并预测中奖号码过程详解
2019/07/09 Python
CSS3中的@keyframes关键帧动画的选择器绑定
2016/06/13 HTML / CSS
交通事故私了协议书
2014/04/16 职场文书
门店业绩提升方案
2014/06/08 职场文书
求职自我评价范文100字
2014/09/23 职场文书
晚自修旷课检讨书怎么写
2014/11/17 职场文书
英语辞职信范文
2015/02/28 职场文书
2015年中个人总结范文
2015/03/10 职场文书
2015年学生会主席工作总结
2015/04/21 职场文书
Python实现socket库网络通信套接字
2021/06/04 Python
使用CSS实现小三角边框原理解析
2021/11/07 HTML / CSS