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