nnabla-ext-cuda

A CUDA Extension of Neural Network Libraries

APACHE-2.0 License

Downloads
3.3K
Stars
92
Committers
52
nnabla-ext-cuda - Version 1.2.0 Release

Published by YasunariZHashimoto about 5 years ago

release-note-bugfix

release-note-build

release-note-op-layer

release-note-utility

Install the latest nnabla by:

pip install nnabla
pip install nnabla-ext-cuda101  # For CUDA version 10.1 users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda101.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla-ext-cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla-ext-cuda101

Note that CUDA 10.1 and cuDNN 7.6 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.6)
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.6)
  • nnabla-ext-cuda101 (CUDA 10.1 x cuDNN 7.6)
pip install nnabla
pip install nnabla-ext-cuda101  # For CUDA 10.1 x cuDNN 7.6 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

The "nnabla-ext-cuda" package is temporarily unavailable. Use of this package is not recommended. Please install nnabla-ext-cuda101, nnabla-ext-cuda100, nnabla-ext-cuda90 or nnabla-ext-cuda80 instead.
The following nnabla CUDA extension packages have been deprecated and the PyPi repository has been closed.

  • nnabla-ext-cuda91
  • nnabla-ext-cuda92
    The following "nnabla-ext-cuda" docker images have been deprecated.
  • py37-cuda92
  • py36-cuda92
  • py27-cuda92
  • py37-cuda92-v1.0.xx
  • py36-cuda92-v1.0.xx
  • py27-cuda92-v1.0.xx

We've decided to change nnabla's versioning policy to semantic versioning.
This change has been applied from version 1.1.0.

nnabla-ext-cuda - Version 1.1.0 Release

Published by YasunariZHashimoto about 5 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla-ext-cuda101  # For CUDA version 10.1 users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda101.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla-ext-cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla-ext-cuda101

Note that CUDA 10.1 and cuDNN 7.6 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.6)
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.6)
  • nnabla-ext-cuda101 (CUDA 10.1 x cuDNN 7.6)
pip install nnabla
pip install nnabla-ext-cuda101  # For CUDA 10. x cuDNN 7.6 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

The "nnabla-ext-cuda" package is temporarily unavailable. Use of this package is not recommended. Please install nnabla-ext-cuda101, nnabla-ext-cuda100, nnabla-ext-cuda90 or nnabla-ext-cuda80 instead.
The following nnabla CUDA extension packages have been deprecated and the PyPi repository has been closed.

  • nnabla-ext-cuda91
  • nnabla-ext-cuda92
    The following "nnabla-ext-cuda" docker images have been deprecated.
  • py37-cuda92
  • py36-cuda92
  • py27-cuda92
  • py37-cuda92-v1.0.xx
  • py36-cuda92-v1.0.xx
  • py27-cuda92-v1.0.xx

We've decided to change nnabla's versioning policy to semantic versioning.
This change has been applied from version 1.1.0.

nnabla-ext-cuda - Version 1.0.20 Release

Published by YasunariZHashimoto over 5 years ago

release-note-op-layer

release-note-utility

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 10.1 and cuDNN 7.6 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.6)
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.6)
  • nnabla-ext-cuda101 (CUDA 10.1 x cuDNN 7.6)
pip install nnabla
pip install nnabla_ext_cuda101  # For CUDA 10. x cuDNN 7.6 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

nnabla-ext-cuda - Version 1.0.19 Release

Published by YasunariZHashimoto over 5 years ago

release-note-bugfix

release-note-build

release-note-op-layer

release-note-utility

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 10.1 and cuDNN 7.6 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.6)
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.6)
  • nnabla-ext-cuda101 (CUDA 10.1 x cuDNN 7.6)
pip install nnabla
pip install nnabla_ext_cuda101  # For CUDA 10. x cuDNN 7.6 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

nnabla-ext-cuda - Version 1.0.18 Release

Published by YasunariZHashimoto over 5 years ago

release-note-break-compat

release-note-bugfix

release-note-build

release-note-op-layer

release-note-utility

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 9.2 and cuDNN 7.4 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.5(Linux), 7.6(Win))
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.5(Linux), 7.6(Win))
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.5(Linux), 7.6(Win))
  • nnabla-ext-cuda101 (CUDA 10.1 x cuDNN 7.5(Linux), 7.6(Win))
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.4 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

nnabla-ext-cuda - Version 1.0.17 Release

Published by YasunariZHashimoto over 5 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 9.2 and cuDNN 7.4 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.5)
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.5)
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.5)
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.4 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

nnabla-ext-cuda - Version 1.0.16 Release

Published by YasunariZHashimoto over 5 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 9.2 and cuDNN 7.4 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.5(win), 7.4(linux))
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.5(win), 7.4(linux))
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.5)
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.4 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

nnabla-ext-cuda - Version 1.0.15 Release

Published by YasunariZHashimoto over 5 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 9.2 and cuDNN 7.3 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.3(win), 7.4(linux))
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.3(win), 7.4(linux))
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.3(win), 7.4(linux))
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.3 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

nnabla-ext-cuda - Version 1.0.14 Release

Published by YasunariZHashimoto over 5 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 9.2 and cuDNN 7.3 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.3(win), 7.4(linux))
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.3(win), 7.4(linux))
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.3(win), 7.4(linux))
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.3 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

nnabla-ext-cuda - Version 1.0.13 Release

Published by YasunariZHashimoto over 5 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 9.2 and cuDNN 7.3 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.3(win), 7.4(linux))
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.3(win), 7.4(linux))
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.3(win), 7.4(linux))
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.3 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

nnabla-ext-cuda - Version 1.0.12 Release

Published by YasunariZHashimoto over 5 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 9.2 and cuDNN 7.3 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.3(win), 7.4(linux))
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.3(win), 7.4(linux))
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.3(win), 7.4(linux))
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.3 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

nnabla-ext-cuda - Version v1.0.11 Release

Published by YasunariZHashimoto over 5 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 9.2 and cuDNN 7.3 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.3)
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.3)
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.3)
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.3 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

nnabla-ext-cuda - Version 1.0.10 Release

Published by YasunariZHashimoto almost 6 years ago

nnabla-ext-cuda - Version 1.0.9 Release

Published by YasunariZHashimoto almost 6 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

Note that CUDA 9.2 and cuDNN 7.3 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.1)
  • nnabla-ext-cuda90 (CUDA 9.0 x cuDNN 7.3)
  • nnabla-ext-cuda92 (CUDA 9.2 x cuDNN 7.3)
  • nnabla-ext-cuda100 (CUDA 10.0 x cuDNN 7.3)
pip install nnabla
pip install nnabla_ext_cuda92  # For CUDA 9.2 x cuDNN 7.3 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

nnabla-ext-cuda - Version 1.0.8 Release

Published by YukioOobuchi about 6 years ago

nnabla-ext-cuda - Version 1.0.7 Release

Published by YasunariZHashimoto about 6 years ago

Install the latest nnabla by:

pip install nnabla
pip install nnabla_ext_cuda  # For CUDA users

Users with python <= 3.4 may experience errors with pip install nnabla and pip install nnabla-ext-cuda.
■ Workaround
Please install matplotlib == 2.2.3 and re-install nnabla, nnabla_ext_cuda.

pip install matplotlib==2.2.3
pip install nnabla
pip install nnabla_ext_cuda

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

nnabla-ext-cuda - Version 1.0.6 Release

Published by YasunariZHashimoto about 6 years ago

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

nnabla-ext-cuda - Version 1.0.5 Release

Published by YasunariZHashimoto about 6 years ago

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

nnabla-ext-cuda - Version 1.0.4 Release

Published by YasunariZHashimoto about 6 years ago

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

Package Rankings
Top 7.17% on Pypi.org
Related Projects