1. ホーム
  2. テンソルフロー

[解決済み】TensorFlowとCUDAのバージョンの組み合わせで互換性があるのはどれ?

2022-04-03 08:04:52

質問

TensorFlowの新しいバージョンの中には、古いCUDAやcuDNNのバージョンと互換性がないものがあることに気づきました。互換性のあるバージョンの概要や、公式にテストされた組み合わせのリストなどは存在するのでしょうか?TensorFlowのドキュメントには見当たりません。

解決方法は?

TL;DR ) この表をご覧ください。 https://www.tensorflow.org/install/source#gpu

一般的に

CUDAのバージョンを確認します。

cat /usr/local/cuda/version.txt

とcuDNNのバージョンです。

grep CUDNN_MAJOR -A 2 /usr/local/cuda/include/cudnn.h

をインストールし、以下の画像にあるような組み合わせ、または ここで .

以下の画像とリンクから、Linux、macOS、Windowsで公式にサポート/テストされているCUDAとTensorFlowの組み合わせの概要を確認することができます。

マイナーな構成です。

以下の仕様は、場合によっては広すぎるかもしれませんので、ここでは、具体的に動作する構成を一つ紹介します。

  • tensorflow-gpu==1.12.0
  • cuda==9.0
  • cuDNN==7.1.4

対応するcudnnをダウンロードすることができます こちら .

テスト済みのビルド構成

をご参照ください。 https://www.tensorflow.org/install/source#gpu は、最新の互換性チャート(公式TFホイール用)をご覧ください。

(数値は2020年5月20日更新)

Linux GPU

LinuxのCPU

macOS GPU

macOSのCPU

Windows GPU

WindowsのCPU

2020年12月5日現在、更新されています。 更新された情報は、こちらをご覧ください。 Linux向けリンク および Windows用リンク .