LoginSignup
101

More than 3 years have passed since last update.

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

Last updated at Posted at 2019-10-08

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を使いましょう

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
101