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 web框架学习笔记
May 03 Python
python中实现将多个print输出合成一个数组
Apr 19 Python
pandas通过loc生成新的列方法
Nov 28 Python
Python字典的核心底层原理讲解
Jan 24 Python
python 定时器,实现每天凌晨3点执行的方法
Feb 20 Python
python 将字符串中的数字相加求和的实现
Jul 18 Python
python生成特定分布数的实例
Dec 05 Python
python 字段拆分详解
Dec 17 Python
在pycharm中为项目导入anacodna环境的操作方法
Feb 12 Python
Keras 利用sklearn的ROC-AUC建立评价函数详解
Jun 15 Python
Python基于time模块表示时间常用方法
Jun 18 Python
Python经常使用的一些内置函数
Apr 11 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中并发读写文件冲突的解决方案
2013/10/25 PHP
php表单请求获得数据求和示例
2014/05/15 PHP
php实现事件监听与触发的方法
2014/11/21 PHP
再Docker中架设完整的WordPress站点全攻略
2015/07/29 PHP
解决form中action属性后面?传递参数 获取不到的问题
2017/07/21 PHP
PHP使用glob方法遍历文件夹下所有文件的实例
2018/10/17 PHP
Yii框架自定义数据库操作组件示例
2019/11/11 PHP
Javascript 倒计时源代码.(时.分.秒) 详细注释版
2011/05/09 Javascript
深入解析contentWindow, contentDocument
2013/07/04 Javascript
javascript字符串替换及字符串分割示例代码
2013/12/12 Javascript
JavaScript中用字面量创建对象介绍
2014/12/31 Javascript
JavaScript中的this关键字使用方法总结
2015/03/13 Javascript
jQuery跨域问题解决方案
2015/08/03 Javascript
js创建数组的简单方法
2016/07/27 Javascript
Bootstrap3 模态框使用实例
2017/02/22 Javascript
vue + elementUI实现省市县三级联动的方法示例
2019/10/29 Javascript
JS中数组实现代码(倒序遍历数组,数组连接字符串)
2019/12/29 Javascript
js绘制一条直线并旋转45度
2020/08/21 Javascript
Python实现Linux中的du命令
2017/06/12 Python
python 判断是否为正小数和正整数的实例
2017/07/23 Python
python中使用正则表达式的后向搜索肯定模式(推荐)
2017/11/11 Python
Python实现螺旋矩阵的填充算法示例
2017/12/28 Python
浅析Python与Mongodb数据库之间的操作方法
2019/07/01 Python
在win64上使用bypy进行百度网盘文件上传功能
2020/01/02 Python
Django Path转换器自定义及正则代码实例
2020/05/29 Python
Nike比利时官网:Nike.com (BE)
2019/02/07 全球购物
有影响力的品牌之家:Our Social Collective
2019/06/08 全球购物
Janie and Jack美国官网:GAP旗下的高档童装品牌
2019/09/09 全球购物
英国羊皮鞋类领先品牌:Just Sheepskin
2019/12/12 全球购物
《动手做做看》教学反思
2014/04/09 职场文书
货物运输服务质量承诺书
2014/05/29 职场文书
毕业实习自我鉴定范文2014
2014/09/26 职场文书
一个都不能少观后感
2015/06/04 职场文书
三好学生主要事迹怎么写
2015/11/03 职场文书
python异步的ASGI与Fast Api实现
2021/07/16 Python
Mysql关于数据库是否应该使用外键约束详解说明
2021/10/24 MySQL