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 IDLE加上自动补全和历史功能
Nov 30 Python
Python列表(list)常用操作方法小结
Feb 02 Python
python入门前的第一课 python怎样入门
Mar 06 Python
基于python神经卷积网络的人脸识别
May 24 Python
Tensorflow加载预训练模型和保存模型的实例
Jul 27 Python
django+xadmin+djcelery实现后台管理定时任务
Aug 14 Python
值得收藏的10道python 面试题
Apr 15 Python
解析Python3中的Import
Oct 13 Python
python3 正则表达式基础廖雪峰
Mar 25 Python
Python 在函数上添加包装器
Jul 28 Python
常用的10个Python实用小技巧
Aug 10 Python
Python脚本实现Zabbix多行日志监控过程解析
Aug 26 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
咖啡的化学
2021/03/03 咖啡文化
第二节--PHP5 的对象模型
2006/11/16 PHP
php的字符串用法小结
2010/06/08 PHP
Javascript客户端脚本的设计和应用
2006/08/21 Javascript
javascript js cookie的存储,获取和删除
2007/12/29 Javascript
JavaScript更改class和id的方法
2008/10/10 Javascript
JavaScript 无符号右移赋值操作
2009/04/17 Javascript
window.event.keyCode兼容IE和Firefox实现js代码
2013/05/30 Javascript
JavaScript Ajax Json实现上下级下拉框联动效果实例代码
2013/11/23 Javascript
关闭ie窗口清除Session的解决方法
2014/01/10 Javascript
JS方法调用括号的问题探讨
2014/01/24 Javascript
window.location 对象所包含的属性
2014/10/10 Javascript
js实现点击文本框显示日期选择器特效代码分享
2020/05/21 Javascript
Javascript获取图片原始宽度和高度的方法详解
2016/09/20 Javascript
js提示框替代系统alert,自动关闭alert对话框的实现方法
2016/11/07 Javascript
NodeJs安装npm包一直失败的解决方法
2017/04/28 NodeJs
详解AngularJS1.6版本中ui-router路由中/#!/的解决方法
2017/05/22 Javascript
详解Nodejs get获取远程服务器接口数据
2019/03/26 NodeJs
前端深入理解Typescript泛型概念
2020/03/09 Javascript
javascript设计模式 ? 解释器模式原理与用法实例分析
2020/04/17 Javascript
解决Vue 给mapState中定义的属性赋值报错的问题
2020/06/22 Javascript
[00:10]DOTA2全国高校联赛 以DOTA2会友
2018/05/30 DOTA
python在windows命令行下输出彩色文字的方法
2015/03/19 Python
Python实现递归遍历文件夹并删除文件
2016/04/18 Python
Python数据结构与算法之图结构(Graph)实例分析
2017/09/05 Python
css3设置box-pack和box-align让div里面的元素垂直居中
2014/09/01 HTML / CSS
HTML5新标签兼容——> 的两种方法
2018/09/12 HTML / CSS
荷兰家电购物网站:Expert.nl
2020/01/18 全球购物
平面网站制作专科生的自我评价分享
2013/12/11 职场文书
肯尼迪就职演说稿
2013/12/31 职场文书
《英英学古诗》教学反思
2014/04/11 职场文书
我的中国心演讲稿
2014/09/04 职场文书
群众路线教育实践活动学习笔记内容
2014/11/06 职场文书
2014年仓库管理工作总结
2014/12/17 职场文书
2015年度招聘工作总结
2015/05/28 职场文书
班组长如何制订适合本班组的工作计划?
2019/07/10 职场文书