索尼开源神经网络库NNabla

发布时间:2017-07-20 16:25:33 | 来源:选自Sony | 作者:佚名 | 责任编辑:胡俊

便携式和多平台

·         Python API 能够在 Linux 和 Windows 上使用。

·         大多数函数库的代码都是使用 C++11 写的,所以可以部署到嵌入设备中。

可扩展性

·         可以像神经网络运算符和优化器那样添加新的模块。

·         该软件库允许开发者添加特定的实现(如在 FPGA 上的实现等)。例如我们提供 CUDA 后端作为扩展,这将可以利用 GPU 加速计算来加速算法。

高效性

·         在单块 CUDA GPU 上速度极快

·         内存优化引擎

·         支持多 GPU(即将可用)

文档

·         https://nnabla.readthedocs.org 

安装

·         https://nnabla.readthedocs.io/en/latest/python/installation.html

入门指南

·         在 tutorial 页面中可以找到很多Jupyter notebook 教程。我们建议由by_examples.ipynb开始在NNabla学习第一个案例,python_api.ipynb文档介绍了NNabla API。

·         我们同样在 examples 中提供了一些其他复杂的案例。 

 

1  2  3  4