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中的对象拷贝示例 python引用传递
Jan 23 Python
Python显示进度条的方法
Sep 20 Python
初步解析Python下的多进程编程
Apr 28 Python
python中文件变化监控示例(watchdog)
Oct 16 Python
python使用pandas实现数据分割实例代码
Jan 25 Python
pandas.dataframe中根据条件获取元素所在的位置方法(索引)
Jun 07 Python
通过 Python 和 OpenCV 实现目标数量监控
Jan 05 Python
Python的赋值、深拷贝与浅拷贝的区别详解
Feb 12 Python
Python基于pandas爬取网页表格数据
May 11 Python
PyCharm vs VSCode,作为python开发者,你更倾向哪种IDE呢?
Aug 17 Python
Python通用唯一标识符uuid模块使用案例
Sep 10 Python
Python 操作 MySQL数据库
Sep 18 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输入输出流学习笔记
2015/05/12 PHP
基于PHP实现用户注册登录功能
2016/10/14 PHP
php利用ZipArchive类操作文件的实例
2020/01/21 PHP
JS控制显示隐藏兼容问题(IE6、IE7、IE8)
2010/04/01 Javascript
js实现图片在未加载完成前显示加载中字样
2014/09/03 Javascript
在JavaScript中构建ArrayList示例代码
2014/09/17 Javascript
jqueryUI里拖拽排序示例分析
2015/02/26 Javascript
jQuery实现伸展与合拢panel的方法
2015/04/30 Javascript
DOM操作原生js 的bug,使用jQuery 可以消除的解决方法
2016/09/04 Javascript
vue实现2048小游戏功能思路详解
2018/05/09 Javascript
vue强制刷新组件的方法示例
2019/02/28 Javascript
JS使用H5实现图片预览功能
2019/09/30 Javascript
安装多版本Vue-CLI的实现方法
2020/03/24 Javascript
[02:57]2014DOTA2国际邀请赛-观众采访
2014/07/19 DOTA
python脚本实现xls(xlsx)转成csv
2016/04/10 Python
python将文本中的空格替换为换行的方法
2018/03/19 Python
python读取和保存图片5种方法对比
2018/09/12 Python
解决pandas.DataFrame.fillna 填充Nan失败的问题
2018/11/06 Python
对pandas将dataframe中某列按照条件赋值的实例讲解
2018/11/29 Python
Django框架实现的简单分页功能示例
2018/12/04 Python
对DJango视图(views)和模版(templates)的使用详解
2019/07/17 Python
flask框架路由常用定义方式总结
2019/07/23 Python
Python unittest框架操作实例解析
2020/04/13 Python
基于python判断字符串括号是否闭合{}[]()
2020/09/21 Python
python实现杨辉三角的几种方法代码实例
2021/03/02 Python
阿根廷票务网站:StubHub阿根廷
2018/04/13 全球购物
adidas澳大利亚官方网站:adidas Australia
2018/04/15 全球购物
广州盈通面试题
2015/12/05 面试题
放飞梦想演讲稿200字
2014/08/26 职场文书
小学教师师德整改措施
2014/09/29 职场文书
2014年创先争优工作总结
2014/12/11 职场文书
综合素质评价个性与发展自我评价
2015/03/06 职场文书
2015年乡镇环保工作总结
2015/04/22 职场文书
详解thinkphp的Auth类认证
2021/05/28 PHP
Python Flask搭建yolov3目标检测系统详解流程
2021/11/07 Python
Java Spring Boot 正确读取配置文件中的属性的值
2022/04/20 Java/Android