keras 指定程序在某块卡上训练实例
场景:某台机器上有三块卡,想同时开三个程序,放到三块卡上去训练。
策略:CUDA_VISIBLE_DEVICES=1 python train.py就可以指定程序在某块卡上训练。
补充知识: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 指定程序在某块卡上训练实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持我们。
相关推荐
-
keras实现多GPU或指定GPU的使用介绍
1. keras新版本中加入多GPU并行使用的函数 下面程序段即可实现一个或多个GPU加速: 注意:使用多GPU加速时,Keras版本必须是Keras2.0.9以上版本 from keras.utils.training_utils import multi_gpu_model #导入keras多GPU函数 import VGG19 #导入已经写好的函数模型,例如VGG19 if G <= 1: print("[INFO] training with 1 GPU...") mod
-
Keras 实现加载预训练模型并冻结网络的层
在解决一个任务时,我会选择加载预训练模型并逐步fine-tune.比如,分类任务中,优异的深度学习网络有很多. ResNet, VGG, Xception等等... 并且这些模型参数已经在imagenet数据集中训练的很好了,可以直接拿过来用. 根据自己的任务,训练一下最后的分类层即可得到比较好的结果.此时,就需要"冻结"预训练模型的所有层,即这些层的权重永不会更新. 以Xception为例: 加载预训练模型: from tensorflow.python.keras.applicat
-
keras多显卡训练方式
使用keras进行训练,默认使用单显卡,即使设置了os.environ['CUDA_VISIBLE_DEVICES']为两张显卡,也只是占满了显存,再设置tf.GPUOptions(allow_growth=True)之后可以清楚看到,只占用了第一张显卡,第二张显卡完全没用. 要使用多张显卡,需要按如下步骤: (1)import multi_gpu_model函数:from keras.utils import multi_gpu_model (2)在定义好model之后,使用multi_gpu
-
Python实现Keras搭建神经网络训练分类模型教程
我就废话不多说了,大家还是直接看代码吧~ 注释讲解版: # Classifier example import numpy as np # for reproducibility np.random.seed(1337) # from keras.datasets import mnist from keras.utils import np_utils from keras.models import Sequential from keras.layers import Dense, Act
-
keras 指定程序在某块卡上训练实例
场景:某台机器上有三块卡,想同时开三个程序,放到三块卡上去训练. 策略:CUDA_VISIBLE_DEVICES=1 python train.py就可以指定程序在某块卡上训练. 补充知识:keras指定GPU及显存使用量 指定GPU import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" 指定GPU和显存使用量 import os from keras.backend.tensorflow_backend import
-
c#语言程序构建基块
目录 1.成员 2.辅助功能 3.字段 4.方法 4.1参数 4.2方法主体和局部变量 4.3静态和实例方法 4.4虚方法.重写方法和抽象方法 4.5方法重载 5.其他函数成员 5.1构造函数 5.2"属性" 5.3索引器 5.4事件 5.5运算符 5.6终结器 6.表达式 7.语句 上文c#语言入门类型和成员中介绍的类型是使用以下构建基块生成的:成员*_.表达式和语句_*. 1.成员 class 的成员要么是静态成员,要么是实例成员. 静态成员属于类,而实例成员则属于对象(类实例).
-
微信小程序实现拍照打卡功能
本文实例为大家分享了微信小程序实现拍照打卡的具体代码,供大家参考,具体内容如下 由于拍照组件是相当于一个块,用隐藏显示的方法不太好,为了更好的用户交互,选择了在一个新的页面调用相机组件,上传图片并保存打卡数据的方式. 小程序端 签到页面wxml <view class="signBtn" bindtap="signSubmit"> <view>{{signTime}}</view> <view>打卡签到<
-
Android编程之在SD卡上进行文件读写操作实例详解
本文实例讲述了Android编程之在SD卡上进行文件读写操作的方法.分享给大家供大家参考,具体如下: 很多知识只有真正理解掌握之后才能运用自如,举一反三.对Java中的文件操作和android系统SD卡里面的文件操作,你觉得有区别吗,显然没有本质区别,如果勉强说有,那也是不足为道滴,但我们在实际运用中却要注意如下几点,不然问题会缠上你. 1.首先想要对android系统SD卡里文件操作需要添加使用权限: android系统是不会让外来程序随意动自己内存的,如果没有许可证,不好意思,不准你动我地盘
-
微信小程序实现文件、图片上传功能
本文实例为大家分享了微信小程序实现文件图片上传的具体代码,供大家参考,具体内容如下 在我看来微信小程序的功能挺强大的,提供了很多API让你直接使用. 这里我说一下微信小程序如何实现图片的上传 1.在微信公众号平台设置uploadFile合法域名 点击设置-开发设置,可以看到服务器域名,点击修改,设置一下你的uploadFile合法域名. 否则会出现以下错误. 2.使用wx.chooseImage和wx.uploadFile实现图片上传 代码如下 wx.chooseImage({ count: 1
-
微信小程序录音实现功能并上传(使用node解析接收)
背景 我在开发小程序的时候,有需求要实现录音功能,并能上传给服务器.小程序录音功能我是使用的微信的wx.getRecorderManager()实现的,通过该方法创建实例,实例录音得到的文件是本地临时文件,上传文件需要使用微信的wx.uploadFile(Object object)方法,这就是本次项目的背景. 小程序端 html页面主要是第一个按钮,两个事件,长按开始录音,松手停止录音.第二个按钮只是一个播放录音的功能,用于确定录音是否成功 <!--pages/record/record.wx
-
keras 自定义loss损失函数,sample在loss上的加权和metric详解
首先辨析一下概念: 1. loss是整体网络进行优化的目标, 是需要参与到优化运算,更新权值W的过程的 2. metric只是作为评价网络表现的一种"指标", 比如accuracy,是为了直观地了解算法的效果,充当view的作用,并不参与到优化过程 在keras中实现自定义loss, 可以有两种方式,一种自定义 loss function, 例如: # 方式一 def vae_loss(x, x_decoded_mean): xent_loss = objectives.binary_
-
小程序实现滑动块效果
本文实例为大家分享了小程序实现滑动块效果的具体代码,供大家参考,具体内容如下 当你在复制的时候 一定要 把js 逻辑的 list 数据更改就行了 小程序的css样式 .box { width: 100vw; background: #F2F2F2; transition: all 3s; } .box-b { height: 8vh; width: 100%; display: flex; justify-content: space-between; background-color: #FA
-
小程序实现日历打卡功能
本文实例为大家分享了小程序实现日历打卡功能的具体代码,供大家参考,具体内容如下 一.效果图展示 老惯例,先上效果图 二.实现思路 1.日历展示 例如下图中: 2021月7月打卡日历页面,共35个日期数据,上月残余4天+本月31天: 2021月6月打卡日历页面,共35个日期数据,上月残余2天+本月30天+下月残余3天: 2021月5月打卡日历页面,共42个日期数据,上月残余6天+本月31天+下月残余5天. [结论]打卡日历页面存在展示35个或42个日期数据的情况,35个或42个日期数据=当前显示月
-
用批处理实现禁止运行指定程序
功能: 1.禁止运行指定程序 2.解除禁止指定程序 3.查看全部禁止程序 复制代码 代码如下: @echo off MODE con COLS=40 LINES=18 title 禁止和解除指定程序运行--陈兵 OO:286569692 :begin setlocal ENABLEDELAYEDEXPANSION cls echo. echo. ╭─────────╮ echo. ╭───┤ 选择如下操作 ├───╮ echo. │ ╰─────────╯ │ echo. │ │ echo. │
随机推荐
- js replace正则相关的诡异问题
- ORACLE常见错误代码的分析与解决(二)
- Javascript 文本框textarea高度随内容自适应增长收缩
- Ajenti开源免费的服务器管理面板和Ajenti V虚拟主机面板及安装与使用详细教程
- cmd SETLOCAL使用介绍
- 教你使用java实现去除各种空格
- iOS中捕获日志与异常示例详解
- Pthon批量处理将pdb文件生成dssp文件
- android图像绘制(三)画布刷屏问题记录
- 用php简单实现加减乘除计算器
- Centos 7下使用RPM包安装MySQL 5.7.9教程
- 一个php+js实时显示时间问题
- linux c 获取本机公网IP的实现方法
- 图片自动缩小 点击放大
- jQuery移除元素自动解绑事件实现思路及代码
- Javascript中的数组常用方法解析
- 浅谈apache和nginx的rewrite的区别
- Android入门之ListView应用解析(二)
- 收集的侃侃的歌46首,照片、侃侃新专辑 下载试听
- python实现k-means聚类算法