Top > blog > 2018-02-13 > TensorFlowをインストール (GPU編)

広告

#blog2navi()
**[[TensorFlowをインストール (GPU編)>blog/2018-02-13/TensorFlowをインストール (GPU編)]] [#t4cf436a]
LEFT:continue form: [[[TensorFlowをインストール>blog/2018-02-13/TensorFlowをインストール]]]

[[TensorFlow:https://www.tensorflow.org/]]のGPUを使わないバージョンのインストールは結構面倒~
インストールの仕方は[[公式ページに書いてある通り:https://www.tensorflow.org/install/install_linux]]

まず必要な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:https://developer.nvidia.com/rdp/cudnn-download]]をインストール。登録が必要。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のバージョンを下げる、自分でコンパイルするなどの方法が必要だと思う

それで解決しても[[前の記事>blog/2018-02-13/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
で直るはず。

[[ソースからコンパイル:https://qiita.com/yijyun-wang/items/12a76667482551eadabc]]

RIGHT:Category: [[[AI>blog/Category/AI]]][[[TensorFlow>blog/Category/TensorFlow]]] - 23:22:01
----
RIGHT:&blog2trackback();
#blog2navi()

広告

リロード   差分   ホーム 一覧 検索 最終更新 バックアップ リンク元   ヘルプ   最終更新のRSS