94
Help us understand the problem. What are the problem?

More than 1 year has passed since last update.

posted at

updated at

Tensorflow GPU, CUDA, CuDNNのバージョン早見表

Tensorflow GPUとその他パッケージのコンパチ早見表

Tensorflow GPU、なぜか動かなくて禿げそう,いやマジではげそう
そんな時は!

下記ページの「テスト済みのビルド設定」を参照されてください。
https://www.tensorflow.org/install/source?hl=ja#gpu_support_2

image.png

上記バージョンに合わせても動かない場合は下記をチェック!

  • 表のとおりにバージョンを合わせたか?(CUDA=9ならば9.1を避けるなど)
  • tensorflowとtensorflow-gpuがダブっていないか?
  • tensorflow-gpuとpython系は同じバージョンでインストールされているか?
  • 動かしたいソースコードはどのバージョンで作られているか?(更新ごとに関数が追加、削除されるため)
  • 最新バージョンは避ける
  • どう考えてもすべて合わせたのに動かない場合は、全体のバージョンを下げてみる
  • GPUとの互換性はあるか? 新しいバージョンと古いGPUは相性が悪い場合がある
  • 何が動かないのか明確化する
    • tensorflow-gpuはgpuを認識しているか?
    • MNISTデータセットを使ったサンプルは動くか?
    • tf系のメソッドの呼び出しでつっかかっているのか?

全ての手を尽くして、寝ても覚めてもダメなら...
nvidia-dockerを使いましょう

Why not register and get more from Qiita?
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away
Sign upLogin
94
Help us understand the problem. What are the problem?