超简单keras 多GPU数据并行

女爷i 2022-04-25 09:40 326阅读 0赞

超级简单：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70][]

import：

from keras.utils import multi_gpu_model

这里以unet为例，首先使用标准的unet建立一个unet模型：

model = unet(input_size=(480, 640, 3))

然后注意自己是否有四个可见的GPU：

os.environ['CUDA_VISIBLE_DEVICES'] = "4,5,6,7"

之后用keras的并行方法包裹已建立模型：

model_simple1 = multi_gpu_model(model, gpus=4)

将包裹好的模型放入热油炸至金黄hhhhh

并行模型model\_simple1直接当做普通的模型compile，fit即可。

注意，这里建议**batchsize最好是能平分到几个GPU上面的，比如四个GPU，你的batchsize要能被4整除，32,28,4,16这样。更多信息：**[https://keras.io/utils/\#multi\_gpu\_model][https_keras.io_utils_multi_gpu_model]

这里提醒一下，使用多gpu建立模型，经过训练保存的权重与单gpu是不一样的，如果你要加载权重，注意加载位置。聪明点

参考资料：[https://keras.io/getting-started/faq/\#how-should-i-cite-keras][https_keras.io_getting-started_faq_how-should-i-cite-keras]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTMyNDk4NTM_size_16_color_FFFFFF_t_70]: /images/20220125/bd0335f46170467a97bd3176717afe28.png
[https_keras.io_utils_multi_gpu_model]: https://keras.io/utils/#multi_gpu_model
[https_keras.io_getting-started_faq_how-should-i-cite-keras]: https://keras.io/getting-started/faq/#how-should-i-cite-keras

发表评论取消回复

表情：

评论列表（有 0 条评论，326人围观）

还没有评论，来说两句吧...

相关阅读

相关 keras 设置GPU使用率

keras 设置GPU使用率代码如下 import tensorflow as tf from keras.backend.tensorflow_backen

ゝ一纸荒年。/ 2023年06月19日 01:57/ 0 赞/ 37 阅读

相关 keras 读取多标签图像数据方式

keras 读取多标签图像数据方式更多python视频教程请到菜鸟教程https://www.piaodoo.com/ 我所接触的多标签数据，主要包括两类： 1、一张图

心已赠人/ 2023年02月18日 08:24/ 0 赞/ 28 阅读

相关 keras调用gpu windows运行环境配置

keras调用gpu windows环境配置及代码变量 CUDA 10.1 ![\[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-dt

秒速五厘米/ 2023年01月20日 09:59/ 0 赞/ 18 阅读

相关 Keras指定GPU训练模式，设置GPU的使用量

最近又要用到深度学习模型训练这块的内容了，之前有些操作都忘记了，这里就想着把这些内容记录一下备忘。我们使用的是阿里云GPU服务器，由于有多个进程同时在训练计算，这里需要限定

缺乏、安全感/ 2022年10月06日 06:47/ 0 赞/ 188 阅读

相关 [keras] 多GPU运行设置/固定权重

\\\\１.所需要的库 `from keras.utils import multi_gpu_model` 参考keras官方文档[multi-gpu][] `ke

- 日理万妓/ 2022年05月13日 04:22/ 0 赞/ 179 阅读

相关超简单keras 多GPU数据并行

超级简单： ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L

女爷i/ 2022年04月25日 09:40/ 0 赞/ 327 阅读

相关超简单的keras函数模型教程

from keras.models import Model from keras.layers import Input, Dense, Lambda

我就是我/ 2022年02月22日 11:08/ 0 赞/ 275 阅读

相关 keras深度训练:GPU设置

keras指定运行时显卡及限制GPU用量固定GPU训练： import os import tensorflow as tf import k

怼烎@/ 2021年12月21日 11:17/ 0 赞/ 324 阅读

相关 [开发技巧]·TensorFlow&Keras GPU使用技巧

\[开发技巧\]·TensorFlow&Keras GPU使用技巧 ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_tex

一时失言乱红尘/ 2021年09月19日 11:14/ 0 赞/ 375 阅读

相关 Keras 数据增强 ImageDataGenerator 多输入多输出

import os os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA

﹏ヽ暗。殇╰゛Y/ 2021年06月24日 15:58/ 0 赞/ 629 阅读