Tensorflow GPUとCUDAのバージョン対応表

GPU版のTensorflowをインストールするときに、Tensorflowのバージョンによって対応しているCUDAやCuDNNのバージョンや、Pythonのバージョンが異なり、上手く動かないということがあります。

バージョン誤りがないかを確認できるように、以下に対応表を作成しましたので、Tensorflowがうまく動かないという人はまず確認してみてください。

Tensorflow GPUとCUDA, CuDNN, Pythonの対応早見表

バージョンの対応は以下の通りとなりますので、インストールしたTensorflow GPUに合ったバージョンになっているか確認してみてください。

Tensorflow GPU VersionPythonCUDAcuDNN
tensorflow-2.6.03.6 – 3.911.28.1
tensorflow-2.5.03.6 – 3.911.28.1
tensorflow-2.4.03.6 – 3.811.08.0
tensorflow-2.3.03.5 – 3.810.17.6
tensorflow-2.2.03.5 – 3.810.17.6
tensorflow-2.1.02.7, 3.5 – 3.710.17.6
tensorflow-2.0.02.7, 3.3 – 3.710.07.4
tensorflow_gpu-1.15.02.7, 3.3 – 3.710.07.4
tensorflow_gpu-1.14.02.7, 3.3 – 3.710.07.4
tensorflow_gpu-1.13.12.7, 3.3 – 3.710.07.4
tensorflow_gpu-1.12.02.7, 3.3 – 3.697
tensorflow_gpu-1.11.02.7, 3.3 – 3.697
tensorflow_gpu-1.10.02.7, 3.3 – 3.697
tensorflow_gpu-1.9.02.7, 3.3 – 3.697
tensorflow_gpu-1.8.02.7, 3.3 – 3.697
tensorflow_gpu-1.7.02.7, 3.3 – 3.697
tensorflow_gpu-1.6.02.7, 3.3 – 3.697
tensorflow_gpu-1.5.02.7, 3.3 – 3.697
tensorflow_gpu-1.4.02.7, 3.3 – 3.686
tensorflow_gpu-1.3.02.7, 3.3 – 3.686
tensorflow_gpu-1.2.02.7, 3.3 – 3.685.1
tensorflow_gpu-1.1.02.7, 3.3 – 3.685.1
tensorflow_gpu-1.0.02.7, 3.3 – 3.685.1

現在の環境にどのバージョンのCUDAやcuDNNが入っているかを確認する方法については、以下の記事にまとめています。

いろいろな深層学習ライブラリを試したりしているときに、そもそも今自分がインストールしているCUDAのバージョンがいくつだっけ……? というこ...

スポンサーリンク

シェアする

  • このエントリーをはてなブックマークに追加

フォローする