どのTensorFlowとCUDAバージョンの組み合わせが互換性がありますか? 質問する

どのTensorFlowとCUDAバージョンの組み合わせが互換性がありますか? 質問する

新しい 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用リンク

おすすめ記事