NVIDIA® cuDNN – GPU Accelerated Machine Learning

[Smorodov 579]

NVIDIA® cuDNN – GPU Accelerated Machine Learning

Key Features:

Forward and backward convolution routines designed for convolutional neural nets, tuned for NVIDIA GPUs

Always optimized for latest NVIDIA GPU architectures

Arbitrary dimension ordering, striding, and subregions for 4d tensors means easy integration into any neural net implementation

Forward and backward paths for many other common layer types (ReLU, Sigmoid, Tanh, pooling, softmax)

Context-based API allows for easy multithreading

Скачать можно здесь: https://developer.nvidia.com/cuDNN