英伟达公司开发了cudnn来帮助大家解决类似于RNN,LSTM,CNN等的速度性能问题,比一般开发人员从零用cuda写要速度快很多倍
因特尔公司相应地开发了mkldnn,这个相当于是CPU版的,而cudnn是GPU版的
因此,如果你希望使用GPU版本的lstm,那就用cudnn来做,如果没有GPU,那就用mkldnn来做
mkldnn 和 mkldnn 的区别
最新推荐文章于 2022-04-24 09:11:58 发布
英伟达公司开发了cudnn来帮助大家解决类似于RNN,LSTM,CNN等的速度性能问题,比一般开发人员从零用cuda写要速度快很多倍
因特尔公司相应地开发了mkldnn,这个相当于是CPU版的,而cudnn是GPU版的
因此,如果你希望使用GPU版本的lstm,那就用cudnn来做,如果没有GPU,那就用mkldnn来做