Skip to content

Version 1.0.6 Release
Compare
Choose a tag to compare
@YasunariZHashimoto YasunariZHashimoto released this 21 Sep 10:52
v1.0.6
7473704
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Learn about vigilant mode.

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Note that CUDA 9.2 and cuDNN 7.2 are set as default if versions are not specified. You can also install the cuda extension with specific versions from one of the following. See also FAQ

  • nnabla-ext-cuda80 (CUDA 8.0 x cuDNN 7.2)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.2)
  • nnabla-ext-cuda91 (CUDA 9.1 x cuDNN 7.1)
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.2)
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.2 users

Additional setup may be required depending on your OS or environment. Please check Python Package Installation Guide for details.

Install Guide

To use C++ inference feature, follow the demonstration on MNIST inference in C++.

Demonstration

For distributed training, you need to build a binary from source. See the guide for building multi-GPU training package.

Build Guide

Assets 2