NNabla是一款用於研究、開發和生產的深度學習框架。NNabla的目標是要能在臺式電腦、HPC叢集、嵌入式裝置和生產伺服器上都能執行。

安裝

安裝NNabla很簡單：

這條命令將安裝NNabla的CPU版本。你還可以通過執行pip install nnabla-ext-cuda來安裝CUDA擴充套件以新增GPU加速。

特點

容易、靈活和富有表現力

基於NNabla C++ 11核心的Python API為你提供了強大的靈活性和高效的生產力。例如，下面的5行程式碼定義了一個具有分類損失的雙層神經網路（超引數用<>括了起來）。

可以通過以下方式來完成訓練：

動態計算圖能夠實現靈活的執行時網路構建。 NNabla可以同時使用靜態和動態圖正規化，並且兩者使用相同的API。

便攜性和多平臺支援

• 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這個例項開始學習。另外，在python_api.ipynb中介紹了NNabla的API。
• 我們還在examples目錄中提供了一些更復雜的例子。