NNabla offers easy extensibility for developers to add new device extensions. The NNabla Python package officially supports the cpu, cuda and cudnn extension, cuda and cudnn extension can dramatically accelerate 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.