TensorFlow多GPU并行的实现
来源:互联网 发布:php导出excel表格乱码 编辑:程序博客网 时间:2024/05/19 00:14
深度学习算法由于数据量非常大、算法复杂度高等特点,常常需要采用并行机制。
常用的并行化深度学习模型训练方式有两种:同步模式和异步模式。
深度模型训练方法
深度学习模型的训练是一个迭代的过程,在每一轮迭代过程中,前向传播算法会根据当前参数的取值,计算出在一小部分训练数据上的预测值,然后反向传播算法,再根据损失函数计算参数的梯度并且更新参数。
异步模式训练方法
在并行化地训练深度学习模型时,不同设备(GPU或CPU)可以再不同训练数据上,运行整个迭代的过程,而不同并行模式的区别在于不同的参数更新方式。
异步模式训练流程图如下:
* 在每一轮迭代时,不同设备会读取参数最新的取值
* 因为设备不同,读取参数取值时间不一样,所以得到的值也可能不一样
* 根据当前参数的取值,和随机获取的一小部分训练数据,不同设备各自运行反向传播的过程,并且独立地更新参数
* 可以认为异步模式,就是单机模式复制了多份,每一份使用不同的训练数据进行训练。
* 在异步模式下,不同设备之前是完全独立的
同步模式训练方法
在同步模式下,所有的设备同时读取参数的取值,并且当反向传播算法完成之后同步更新参数的取值,单个设备不会单独对参数进行更新,而会等所有设备都完成反向传播之后再统一更新参数。
同步模式训练流程图如下:
* 图中在迭代每一轮时,不同设备首先统一读取当前参数的取值,并随机获取一小部分数据
* 然后在不同设备上运行反向传播过程得到在各自训练数据上的参数的梯度
* 注意:虽然所有设备使用的参数是一致的,但是因为训练数据不同,所以得到的参数的梯度可能不一样
* 当所有设备完成反向传播的计算之后,需要计算出不同设备上参数梯度的平均值
* 最后再根据平均值对参数进行更新
同步/异步优劣比较
- 同步模式解决了异步模式中存在的参数更新问题,然而同步模式的效率却低于异步模式
- 在同步模式下,每一轮迭代都需要设备统一开始、统一结束
- 如果设备的运行速度不一致,那么每一轮训练都需要等待最慢的设备结束才能开始更新参数,于是很多时间将被花在等待上
- 虽然理论上异步模式存在缺陷,但是因为训练深度学习模型时,使用的随机梯度下降本身就是梯度下降的一个近似解法,而且即使是梯度下降也无法保证达到全局最优
- 所以在实际应用中,相同时间内,使用异步模式训练的模型不一定比同步模式差
代码示例
#将神经网络的优化过程跑在不同的GPU上for i in range(N_GPU): with tf.debice('/gpu:%d'%i) with tf.name_scope('GPU_%d'%i) as scope: cur_loss = get_loss(x,y_regularizer,scope) #tf.get_variable的命名空间 tf.get_variable_scope().reuse_variables() #使用当前gpu计算所有变量的梯度 grads= opt.compute_gradients(cur_loss) tower_grads.append(grads)#计算变量的平均梯度grads = average_gradients(tower_grads)#使用平均梯度更新参数apply_gradient_op = opt.apply_gradients(grads,global_step = global)
- TensorFlow多GPU并行的实现
- TensorFlow多GPU并行的实现
- 使用Tensorflow实现多GPU并行训练
- TensorFlow多GPU并行计算
- tensorflow多gpu并行计算
- 分享一下三个写的比较好的tensorflow多GPU模型,实现数据并行/模型并行
- TensorFlow 多 GPU 处理并行数据
- Tensorflow实战学习(四十)【多GPU并行】
- tensorflow43《TensorFlow实战》笔记-09-02 多GPU并行 code
- 深度学习TensorFlow如何使用多GPU并行模式?
- Tensorflow深度学习之十八:多GPU并行
- 89、tensorflow使用GPU并行计算
- 基于GPU实现的高效的并行数据结构
- 卷积操作的GPU粗粒度并行实现及测试
- Tensorflow-gpu的错误
- tensorflow25《TensorFlow实战Google深度学习框架》笔记-10-02 多GPU并行 code
- Tensorflow多GPU
- 多GPU Tensorflow
- 如何把.npy格式与PIL image相互转换
- Gradle依赖子模块配置
- 【汇编】谈谈volatile变量
- layui参数
- MDK 4.7a自动补齐没了?
- TensorFlow多GPU并行的实现
- HTML和CSS入门(1)
- dom4j处理xml在linux环境下中文乱码
- cardinality feedback used for this statement 的BUG
- 序列所有可能的出栈顺序
- Java操作linux脚本
- Linux——信号掩码(signal mask)
- Maven构建SSM框架JavaWeb项目pom.xml文件整理
- leetcode 41|268. First Missing Positive 268. Missing Number