Tensorflow GPUとその他パッケージのコンパチ早見表
Tensorflow GPU、なぜか動かなくて禿げそう,いやマジではげそう
そんな時は!
下記ページの「テスト済みのビルド設定」を参照されてください。
https://www.tensorflow.org/install/source?hl=ja#gpu_support_2
上記バージョンに合わせても動かない場合は下記をチェック!
- 表のとおりにバージョンを合わせたか?(CUDA=9ならば9.1を避けるなど)
- tensorflowとtensorflow-gpuがダブっていないか?
- tensorflow-gpuとpython系は同じバージョンでインストールされているか?
- 動かしたいソースコードはどのバージョンで作られているか?(更新ごとに関数が追加、削除されるため)
- 最新バージョンは避ける
- どう考えてもすべて合わせたのに動かない場合は、全体のバージョンを下げてみる
- GPUとの互換性はあるか? 新しいバージョンと古いGPUは相性が悪い場合がある
- 何が動かないのか明確化する
- tensorflow-gpuはgpuを認識しているか?
- MNISTデータセットを使ったサンプルは動くか?
- tf系のメソッドの呼び出しでつっかかっているのか?
全ての手を尽くして、寝ても覚めてもダメなら...
nvidia-dockerを使いましょう