やおよろず

新社会人が毎日更新します。

ChainerX でエラーが出た時の対処法

qiita.com

この記事を参考にChainerを導入しようとしたところ、サンプルコードのtrain_mnist.py で躓いた。

ModuleNotFoundError: No module named 'chainerx'


どうやらChainerXとやらが入っていないぞと。


ChainerXとはなんぞや

github.com

こちらを参照のこと。

つまり環境変数をいじってからプレリリース版を入れようねと。

具体的には、以下の用に変更する。

CHAINER_BUILD_CHAINERX=1
MAKEFLAGS=-j8


変更方法はWin版では "set", Macでは"export"。
環境変数をいじった後にpipでインストール。

$ pip install --pre chainer

環境変数をいじったのに

以上のことを試しても自分はダメだった為、一回再インストールすることに。

$ pip uninstall chainer


そして上記の方法を試した所無事インストールできました。成功したか判別方法は以下のコマンドで確認できます

>>> import chainer
>>> chainer.__version__
'6.0.0b1'


これでtrain_mnist.pyが起動できました。

追記

起動はできたもののエラーが起こった。
エラーが起きた個所のtrain_mnist.pyのソースコードを読むと、

# TODO(niboshi): Temporarily disabled for chainerx. Fix it.


とのことでした...