Top > blog > 2018-02-13

広告

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をインストール。登録が必要。debファイルが配布されているのでそれを使ってインストール

後は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

で直るはず。

ソースからコンパイル

Category: [AI][TensorFlow] - 23:22:01

TensorFlowをインストール

ディープラーニング用のライブラリであるTensorFlowをインストールしてみた。

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

ここでは簡単にGPUを使わないバージョンをpipを使った方法でのインストールに付いて書いてみる。

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

Python 2.7

# apt-get install python-pip python-dev

Python 3

# apt-get install python3-pip python3-dev

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

Python 2.7

# pip install tensorflow

Python 3

# pip3 install 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

で直るはず。

GPUを使うバージョンはもう少し複雑なので後で書いてみる

Category: [AI][TensorFlow] - 23:03:07


広告

リロード   差分   ホーム 一覧 検索 最終更新 バックアップ リンク元   ヘルプ   最終更新のRSS
Last-modified: Tue, 13 Feb 2018 23:03:07 JST (2235d)