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学习笔记之open()函数打开文件路径报错问题
Apr 28 Python
Python Dataframe 指定多列去重、求差集的方法
Jul 10 Python
Python实现的建造者模式示例
Aug 06 Python
对numpy中二进制格式的数据存储与读取方法详解
Nov 01 Python
python3使用pandas获取股票数据的方法
Dec 22 Python
Python写一个基于MD5的文件监听程序
Mar 11 Python
python中时间模块的基本使用教程
May 14 Python
python经典趣味24点游戏程序设计
Jul 26 Python
python Kmeans算法原理深入解析
Aug 23 Python
Python 私有属性和私有方法应用场景分析
Jun 19 Python
Tensorflow使用Anaconda、pycharm安装记录
Jul 29 Python
python 将Excel转Word的示例
Mar 02 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
收藏的一个php小偷的核心程序
2007/04/09 PHP
Wordpress 相册插件 NextGEN-Gallery 添加目录将中文转为拼音的解决办法
2010/12/29 PHP
php中利用post传递字符串重定向的实现代码
2011/04/21 PHP
PHP利用APC模块实现文件上传进度条的方法
2015/01/26 PHP
php正则匹配文章中的远程图片地址并下载图片至本地
2015/09/29 PHP
JQuery文本框高亮显示插件代码
2011/04/02 Javascript
js自执行函数的几种不同写法的比较
2012/08/16 Javascript
Jquery实现的角色左右选择特效
2014/05/21 Javascript
javascript将url中的参数加密解密代码
2014/11/17 Javascript
文字垂直滚动之javascript代码
2015/07/29 Javascript
HTML5 实现的一个俄罗斯方块实例代码
2016/09/19 Javascript
BootStrap 动态添加验证项和取消验证项的实现方法
2016/09/28 Javascript
JS填写银行卡号每隔4位数字加一个空格
2016/12/19 Javascript
Javascript面试经典套路reduce函数查重
2017/03/23 Javascript
Bootstrap Table使用整理(二)
2017/06/09 Javascript
微信小程序滚动Tab实现左右可滑动切换
2017/08/17 Javascript
JQuery 又谈ajax局部刷新
2017/11/27 jQuery
用Fundebug插件记录网络请求异常的方法
2019/02/21 Javascript
详解小程序中h5页面onShow实现及跨页面通信方案
2019/05/30 Javascript
layui 选择列表,打勾,点击确定返回数据的例子
2019/09/02 Javascript
js中火星坐标、百度坐标、WGS84坐标转换实现方法示例
2020/03/02 Javascript
Node.js API详解之 os模块用法实例分析
2020/05/06 Javascript
原生JavaScript实现换肤
2021/02/19 Javascript
python封装对象实现时间效果
2020/04/23 Python
Python爬虫工程师面试问题总结
2018/03/22 Python
tensorflow实现加载mnist数据集
2018/09/08 Python
python+opencv3.4.0 实现HOG+SVM行人检测的示例代码
2021/01/28 Python
如何使用canvas绘制可移动网格的示例代码
2020/12/14 HTML / CSS
国际化的太阳镜及太阳镜配件零售商:Sunglass Hut
2016/07/26 全球购物
Ralph Lauren意大利官方网站:时尚界最负盛名的品牌之一
2018/10/18 全球购物
俄罗斯女装店:12storeez
2019/10/25 全球购物
Happy Socks英国官网:购买五颜六色的袜子
2020/11/03 全球购物
网吧管理制度范本
2015/08/05 职场文书
Nginx代理同域名前后端分离项目的完整步骤
2021/03/31 Servers
Python基础之函数嵌套知识总结
2021/05/23 Python
opencv 分类白天与夜景视频的方法
2021/06/05 Python