NNabla分散学習用のPythonパッケージ

2018年9月28日 金曜日

Uncategorized

Posted by Kazuki Yoshiyama

NNablaで分散学習が可能なパッケージ(python3.5 and python3.6)をリリースしました.次のコマンドでインストールできます.

pip install nnabla-ext-cuda92-nccl2-ubuntu16

すでにNNablaが入っている場合は,NNablaを削除してください.もしくは,まっさら環境をAnaconda, virtualenv, またはpyenv等で作成してからインストールしてください.

名前の通り,pipでインストール可能なのは今のところubuntu16.04のみですが,分散学習も可能なPythonパッケージは,様々なLinux,いろんなCPUアーキテクチャ(Intel and Power)でもビルド&インストール可能です.詳しくは,ドキュメントをご覧ください.