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实现动态添加类的属性或成员函数的解决方法
Jul 16 Python
举例讲解Python中的list列表数据结构用法
Mar 12 Python
Python2.7简单连接与操作MySQL的方法
Apr 27 Python
Python实现学校管理系统
Jan 11 Python
python 将大文件切分为多个小文件的实例
Jan 14 Python
Python实现的企业粉丝抽奖功能示例
Jul 26 Python
Django项目主urls导入应用中views的红线问题解决
Aug 10 Python
django model的update时auto_now不被更新的原因及解决方式
Apr 01 Python
利用Python制作动态排名图的实现代码
Apr 09 Python
Python Selenium截图功能实现代码
Apr 26 Python
Django实现drf搜索过滤和排序过滤
Jun 21 Python
Python获取江苏疫情实时数据及爬虫分析
Aug 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之第四天
2006/10/09 PHP
PHP+javascript模拟Matrix画面
2006/10/09 PHP
php加速器eAccelerator的配置参数、API详解
2014/05/05 PHP
PHP实现简单数字分页效果
2015/07/26 PHP
PHP基于socket实现客户端和服务端通讯功能
2017/07/13 PHP
thinkphp5 框架结合plupload实现图片批量上传功能示例
2020/04/04 PHP
Javascript 各浏览器的 Javascript 效率对比
2008/01/23 Javascript
javascript实现playfair和hill密码算法
2014/12/07 Javascript
javascript编写实用的省市选择器
2015/02/12 Javascript
JS代码防止SQL注入的方法(超简单)
2016/04/12 Javascript
详细探究ES6之Proxy代理
2016/07/22 Javascript
Node.JS中事件轮询(Event Loop)的解析
2017/02/25 Javascript
vue-cli中的webpack配置详解
2017/09/25 Javascript
Angular学习教程之RouterLink花式跳转
2018/05/03 Javascript
js实现input密码框显示/隐藏功能
2020/09/10 Javascript
nodejs同步调用获取mysql数据时遇到的大坑
2019/03/02 NodeJs
基于vue.js实现购物车
2020/01/15 Javascript
es6中class类静态方法,静态属性,实例属性,实例方法的理解与应用分析
2020/02/15 Javascript
js实现的订阅发布者模式简单示例
2020/03/14 Javascript
JS highcharts实现动态曲线代码示例
2020/10/16 Javascript
Python获取脚本所在目录的正确方法
2014/04/15 Python
深入浅析Python中list的复制及深拷贝与浅拷贝
2018/09/03 Python
PyCharm配置mongo插件的方法
2018/11/30 Python
pytorch绘制并显示loss曲线和acc曲线,LeNet5识别图像准确率
2020/01/02 Python
python logging 日志的级别调整方式
2020/02/21 Python
python如何保存文本文件
2020/06/07 Python
德国珠宝和配件商店:Styleserver
2021/02/23 全球购物
十岁生日父母答谢词
2014/01/18 职场文书
公司年会策划方案
2014/05/17 职场文书
终止或解除劳动合同及劳动关系的证明书
2014/10/06 职场文书
司考复习计划
2015/01/19 职场文书
校车司机安全责任书
2015/05/11 职场文书
搞笑婚礼主持词开场白
2015/11/24 职场文书
创业计划书之美甲店
2019/09/20 职场文书
mongodb的安装和开机自启动详细讲解
2021/08/02 MongoDB
聊聊CSS粘性定位sticky案例解析
2022/06/01 HTML / CSS