索尼開源神經網絡庫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 中提供了一些其他復雜的案例。