cudnn可以给rnn加速吗
发布网友
发布时间：2022-04-22 19:15
共2个回答
热心网友
时间：2023-06-25 01:14
可以
默认你说的gpu加速是指NVIDIA的cuda加速
CPU是*处理单元，gpu是图形处理单元
简单的说，gpu由上千个流处理器(core)作为运算器。执行采用单指令多线程(SIMT)模式。
在cuda编程中有grids、blocks、threads三个维度，大量threads组成一个block，大量blocks组成一个grid。每个block内的所有threads执行相同的操作。
在gpu上执行的函数叫kernel函数，用__device__或__global__声明
热心网友
时间：2023-06-25 01:15
可以的，keras框架就有个叫cudnnlstm的lstm的版本,你可以试一下：
网页链接在这个文档里搜关键字cudnnlstm
