Public | Automated Build

Last pushed: 11 days ago
Short Description
Docker Image of NNabla with CUDA extension.
Full Description

A CUDA Extension of Neural Network Libraries

This repository provides an official CUDA/cuDNN-accelerated extension of the
Neural Network Libraries deep learning framework.

In order to use it, the default context needs to be changed from 'cpu' to
'cuda.cudnn':

from nnabla.contrib.context import extension_context

ctx = extension_context('cuda.cudnn', device_id=args.device_id)
nn.set_default_context(ctx)

Currently, the installation documentation and usage is integrated into the base
NNabla.

FAQ

  • Coming soon
Docker Pull Command
Owner
nnabla
Source Repository