NNabla offers easy extensibility for developers to add new device extensions. The NNabla Python package officially supports the cudnn extension, which dramatically accelerates computation by leveraging NVIDIA CUDA GPUs with cuDNN computation primitives.
You can manually import extensions by:
import nnabla_ext.cudnn
See :ref:`python-package-installation` to install the CUDA extension.nnabla.ext_utils
list_extensions
import_extension_module
get_extension_context
All extension modules must have the following functions.
TODO: Link to a page of how to create a new extension.