GPUパソコンでTensorFlowをインストールすると時にハマった件
今回GPU付きPCにTensorFlowをインストールしようとしました。
環境はWin10です。その時に失敗した2点を明確にしておきます。
TensorFlowとCUDAのバージョン
二つの表をご覧ください。最新バージョンと対応CUDAが異なっています。当然英語版の方が最新です。
一つは日本語版、一つは英語版です。
どちらもhttps://www.tensorflow.org/install/source_windows
の下の方に書いてあります。日本語版も英語版もURLが同じなので注意が必要です。
Language を英語に切り替えましょう
Anaconda仮想環境でのTensorFlowのインストール
conda install tensorflow-gpuでインストールすると下記のエラーがでます。
GPU sync failed
原因はcondaでインストールしたためです。pipでインストールしましょう。
https://fight-tsk.blogspot.com/2016/11/anacondatensorflowgpu.html
このサイトで情報を得ました。
ここから引用します。
Only the CPU version of TensorFlow is available at the moment and can be installed in the conda environment for Python 2 or Python 3.
Pyhon2/3 conda環境内でインストールされたTensorFlowは現時点ではCPUのみが有効です。
残念ながらtensorflowの公式ページで上記の文面を発見することはできませんでした。