blog/2018-02-13/TensorFlowをインストール (GPU編) のバックアップ(No.1)

• バックアップ一覧
• 差分 を表示
• 現在との差分 を表示
• 現在との差分 - Visual を表示
• ソース を表示
• blog/2018-02-13/TensorFlowをインストール (GPU編) へ行く。
  • 1 (2018-02-13 (火) 23:23:05)

TensorFlowをインストール (GPU編)

TensorFlowのGPUを使わないバージョンのインストールは結構面倒
インストールの仕方は公式ページに書いてある通り

まず必要なpipなどのパッケージをインストール

Python 2.7

# apt-get install python-pip python-dev

Python 3

# apt-get install python3-pip python3-dev

CUDA関連のパッケージををインストール

• nvidia-cuda-toolkit
• libcupti-dev など

cuDNNをインストール。登録が必要

後はpipでインストールすればいい

Python 2.7

# pip install tensorflow-gpu

Python 3

# pip3 install tensorflow-gpu

で適当なテストプログラムなどを実行すると

ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

というエラーが出る場合はCUDAのバージョンが合ってないらしい。
「apt-get upgrade」や「apt-get dist-upgrade」をしてみると大丈夫なときもある。
それでもだめなら「experimental」を使うかTensorFlowのバージョンを下げる、自分でコンパイルするなどの方法が必要だと思う

それで解決しても前の記事と同様に

RuntimeError: module compiled against API version 0xc but this version of numpy is 0xa
ImportError: numpy.core.multiarray failed to import
ImportError: numpy.core.umath failed to import
ImportError: numpy.core.umath failed to import
2018-02-13 22:50:21.031602: F tensorflow/python/lib/core/bfloat16.cc:664] Check failed: PyBfloat16_Type.tp_base != nullptr 
中止 (コアダンプ)

みたいなエラーがでるときがある。これはnumpyのバージョンが合っていないため。
これを回避するには新しいnumpyをインストールする

Python 2.7

# pip install numpy

Python 3

# pip3 install numpy

で直るはず。