为什么keras LSTM使用GPU无法加速？

最新推荐文章于 2024-04-01 23:05:53 发布
最新推荐文章于 2024-04-01 23:05:53 发布
阅读量6k 收藏 14
点赞数
文章标签： gpu last Keras
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_35834894/article/details/100034755
版权

在做实验的过程中，如果不用gpu的话，我的lstm跑一遍的时间是2分钟半，

在网上找了好多资料，但是中文网也没人说这个问题，然后我以为是因为keras在使用gpu上出了问题，

但是经过测试，我所使用的平台也有gpu

那么就是keras以及网络自身的问题了

然后我找到了一个回答

那就是用CuDNNLSTM代替LSTM

keras官方文档

Fast LSTM implementation with CuDNN.
Can only be run on GPU, with the TensorFlow backend.
hhggggghhh
关注
  • 0
    点赞
  • 踩
  • 14
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
[Python人工智能] 二十四.易学智能GPU搭建Keras环境实现LSTM恶意URL请求分类
杨秀璋的专栏
11-15 4794
这篇文章简单讲解易学智能GPU搭建Keras环境的过程，并实现了LSTM文本分类的实验。本专栏主要结合作者之前的博客、AI经验和相关视频及论文介绍，后面随着深入会讲解更多的Python人工智能案例及应用。基础性文章，希望对您有所帮助，如果文章中存在错误或不足之处，还请海涵~作者作为人工智能的菜鸟，希望大家能与我在这一笔一划的博客中成长起来。写了这么多年博客，尝试第一个付费专栏，但更多博客尤其基础性文章，还是会继续免费分享，但该专栏也会用心撰写，望对得起读者，共勉！
keras bi-lstm_LSTM用于文本生成的应用介绍-使用Keras和启用GPU的Kaggle Kernels
07-24 499
keras bi-lstmby Megan Risdal 梅根·里斯达尔(Megan Risdal) LSTM用于文本生成的应用介绍-使用Keras和启用GPU的Kaggle Kernels (An applied introduction to LSTMs for text generation — using Keras and GPU-enabled Kaggle Kernels) Ka...
神经网络算法：一文搞懂LSTM（长短期记忆网络）
最新发布
2401_84033492的博客
04-01 3687
本文将从LSTM的本质、LSTM的原理、LSTM的应用三个方面，带您一文搞懂长短期记忆网络Long Short Term Memory | LSTM。
LSTM+Keras
Provencess的博客
10-13 171
LSTM层 keras.layers.recurrent.LSTM(units, activation='tanh', recurrent_activation='hard_sigmoid', use_bias=True, kernel_initializer='glorot_uniform', recurrent_initializer='orthogonal', bias_initializ...
解决keras不能用GPU加速的问题。
weixin_43917589的博客
04-18 2021
1、keas如果搭配Tensorflow1.x用，就一定要用先安装tensorflow-gpu、然后安装keras-gpu。 在这里插入代码片 conda install tensorflow-gpu==1.15 conda install keras-gpu==2.3.1 不能安装keras，而是要安装keras-gpu。因为Tensorflow1.x是区分CPU版本和GPU版本的。 2、keras如果搭配Tensorflow2.x用，则keras不必区分CPU版本和GPU版本。 具体keras和ten
tensorflow2.0 lstm/gru无法使用GPU加速
guli_qi的博客
05-11 1232
https://colab.research.google.com/github/tensorflow/docs/blob/master/site/en/guide/keras/rnn.ipynb#scrollTo=76xAs7epaX21
唐诗，藏头诗，按需自动生成古诗，基于Keras、LSTM-RNN。文档齐全
05-31
唐诗，古诗，五言绝句自动生成，基于Keras，LSTM-RNN。 附带训练好的模型文件，可直接上手使用。 功能：藏头诗，随机写诗，给定第一句诗/字进行作诗 测试结果： 在GPU Tesla K80上，2s/epoch，一共有3w+个epoch。...
LSTM：使用LSTM对嘈杂的语音文件进行语音活动检测。 LSTM是通过Keras实现的。 数据处理使用Python，MATLAB和Bash完成。 实验在Johns Hopkins CLSP GPU上完成
02-02
LSTM 使用LSTM对嘈杂的语音文件进行语音活动检测。 LSTM是通过Keras实现的。 数据处理使用Python，MATLAB和Bash完成。 实验是在Johns Hopkins CLSP GPU上完成的。
LSTM工具箱
02-04
LSTM工具箱 最快的Matlab RNN库之一。 性能 模型：一个LSTM模型具有[1024,1024,1024]隐藏尺寸和10个时间步长，并具有256个暗淡输入。 设备：i7-4710hq，GTX940m LSTMtoolbox：60sec / epoch Keras（1.2.2，...
nmt-keras：使用Keras进行神经机器翻译
02-06
NMT-凯拉斯 Keras的神经机器翻译。 图书馆文件： 注意递归神经网络NMT模型变压器NMT模型功能（除了完整的Keras宇宙外）：。 :red_exclamation_mark: 多GPU训练（仅适用于Tensorflow）。 。 。 在线学习和交互式神经...
Keras-Tutorials:使用Keras框架进行深度学习简介
05-19
使用以下命令运行非GPU容器： docker run -it -p 8888:8888 keras-notebookDocker镜像扩展了正式的。 当前，该映像不支持GPU容器，并且不在Docker Hub上（即将更新）。话题递归神经网络（LSTM） Word2vec 转移学习...
【机器学习】LSTM算法原理及加速算法
kouqikamalu的博客
06-03 2426
LSTM算法介绍 这里有一本书，是由Jason Brownlee所著《Long Short Term Memory Networks with Python》，里面详细 介绍了lstm相对于mlp的优势及前向后向算法。 下面给出我的csdn链接（感谢支持）和百度云链接，大家按需自取就好。 https://download.csdn.net/download/kouqikamalu/19359306 链接：https://pan.baidu.com/s/1kT0KAGGNew3BkFByi6os2A
keras终止训练后显存不释放_GPU训练加速原理（附KerasGPU训练技巧）
weixin_39678103的博客
11-23 497
0.深入理解GPU训练加速原理GPU是如何加速的呢？我打算从两个方面来解答：单个GPU较于CPU加速：在训练网络中，其实大量的运算资源都消耗在了数值计算上面，大部分网络训练的过程都是1.计算loss，2.根据loss求梯度，3.再根据梯度更新参数(梯度下降原理)。无论在GPU还是CPU中，都是不断重复123步。但是由于CPU是通用计算单元(并不擅长数值运行)，而GPU特长是图像处理(数值计算)。所...
LSTM在Imagination神经网络加速器上的运行
weixin_49393016的博客
02-23 236
近年来，语音识别变得越来越重要：它使计算机能够将口语翻译成文本。它可以在不同类型的应用中找到，例如翻译器或隐藏字幕。以Mozilla的DeepSpeech 为例，它是一个开源的语音转文本引...
RNN、LSTM、GRU等模型使用GPU时的常见错误
weixin_44457930的博客
01-24 3153
错误信息： RuntimeError: Input and hidden tensors are not at the same device, found input tensor at cuda:0 and hidden tensor at cpu 错误提示说，输入的张量在cuda中，但模型的隐藏层在CPU中，但我们检查模型，发现模型已经被放入到了cuda中了 # 实例化模型 model = LSTM(input_dim=input_dim, hidden_dim=hidden_dim, output
『开发技术』GPU训练加速原理（附KerasGPU训练技巧）
简明AI工作室
08-13 6686
0.深入理解GPU训练加速原理 我们都知道用GPU可以加速神经神经网络训练（相较于CPU），具体的速度对比可以参看我之前写的速度对比博文：[深度应用]·主流深度学习硬件速度对比（CPU，GPU，TPU） GPU是如何加速的呢？ 我打算从两个方面来解答： 单个GPU较于CPU加速： 在训练网络中，其实大量的运算资源都消耗在了数值计算上面，大部分网络训练的过程都是1.计算loss，2.根据l...
模型训练中，不调用gpu比调用gpu更快的情况
永远的我们的博客
09-29 2133
gpu，cpu，训练时间
lstm网络一般训练多少轮_利用LSTM(长短期记忆网络)来处理脑电数据
weixin_39763683的博客
11-28 1916
文章来源于公众号"脑机接口社区"利用LSTM(长短期记忆网络)来处理脑电数据​mp.weixin.qq.comRose小哥今天介绍一下用LSTM来处理脑电数据。LSTMs(Long Short Term Memory networks，长短期记忆网络)简称LSTMs，很多地方用LSTM来指代它。本文也使用LSTM来表示长短期记忆网络。LSTM是一种特殊的RNN网络(循环神经网络)。想要说清楚LST...
Keras中 LSTM 模型解读
pengxiang1998的博客
09-23 3200
t时刻的输入X和t-1时刻的输出h(t-1)进行拼接，然后输入cell中，其实可以这样理解，我们的输入X(t)分别feed进了四个小黄矩形中，每个小黄矩形中进行的运算和正常的神经网络的计算一样（矩阵乘法），有关记忆的部分完全由各种门结构来控制（就是0和1），同时在输入时不仅仅有原始的数据集，同时还加入了上一个数据的输出结果，也就是。关于return_sequence的理解：即返回序列，若为true，则说明后面还有LSTM模型需要接收序列，即存在步长，若为false，说明不需要接收模型，则没有步长这个参数。
如何使用GPU训练LSTM模型
05-05
使用GPU训练LSTM模型需要安装CUDA和cuDNN，这是NVIDIA提供的用于GPU加速的工具。安装后，您可以使用深度学习框架（如TensorFlow或PyTorch）来编写LSTM模型，并将其配置为在GPU上进行训练。 以下是使用TensorFlow在GPU上训练LSTM模型的简单步骤： 1. 安装CUDA和cuDNN。 2. 安装TensorFlow并确保它可以访问GPU。您可以使用以下命令验证： ``` import tensorflow as tf tf.test.is_gpu_available() ``` 3. 加载数据集并准备数据。这通常涉及将数据转换为张量，并将其拆分为训练集和测试集。 4. 构建LSTM模型。您可以使用TensorFlow中的 `tf.keras.layers.LSTM` 层来构建模型。 5. 编译模型并指定优化器和损失函数。 6. 调用 `model.fit()` 函数来训练模型，将 `epochs` 参数设置为所需的训练次数。 以下是一个简单的示例代码，演示了如何使用TensorFlow在GPU上训练LSTM模型： ``` import tensorflow as tf from tensorflow.keras.layers import LSTM, Dense # 加载数据集并准备数据 # ... # 构建模型 model = tf.keras.Sequential([ LSTM(64, input_shape=(None, input_dim)), Dense(output_dim, activation='softmax') ]) # 编译模型 model.compile(optimizer='adam', loss='categorical_crossentropy') # 在GPU上训练模型 with tf.device('/GPU:0'): model.fit(x_train, y_train, epochs=10, batch_size=32) ``` 请注意，训练LSTM模型需要大量的计算资源和时间。使用GPU可以显著加速训练过程，但仍需要耐心等待模型训练完成。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值