技术文档

cuDNN

cuDNN为专门为深度学习设计的一套GPU加速库,很好的实现了深度学习中常见的一些内容,如卷积层、池化层、归一化层和激活函数等,cuDNN是NVIDIA深度学习开发工具的一部分。
深度学习研究人员和框架的开发人员高度依赖cuDNN,用于GPU有效加速。cuDNN让他们能够花更多的时间精力在网络训练和软件应用的开发上,而不是耗费时间在底层GPU性能调节上。cuDNN被运用于众多的深度学习框架中,其中包括Caffe2, MATLAB, Microsoft Cognitive Toolkit, TensorFlow, Theano, 和 PyTorch等。

cuDNN支持的深度学习架构:

                         

 
 
                nbsp;          

                            

 
 
附录: