新力開源神經網路庫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 中提供了一些其他複雜的案例。