新しい TensorFlow バージョンの一部は、古い CUDA および cuDNN バージョンと互換性がないことに気付きました。互換性のあるバージョンの概要や、公式にテストされた組み合わせのリストはありますか? TensorFlow のドキュメントでは見つかりません。
ベストアンサー1
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用リンク。