索尼开源神经网络库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 中提供了一些其他复杂的案例。