web-dev-qa-db-ja.com

互換性のあるTensorFlowとCUDAのバージョンの組み合わせはどれですか？

私はいくつかの新しいTensorFlowバージョンが古いCUDAとcuDNNバージョンと互換性がないことに気づきました。互換性のあるバージョンの概要、あるいは正式にテストされた組み合わせのリストさえ存在しますか？ TensorFlowのドキュメントには見つかりません。

tensorflowcudaversioncompatibilitycudnn
55
Fábio

一般に

CUDAのバージョンを確認してください。

cat /usr/local/cuda/version.txt

とcuDNNのバージョン：

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

そして、以下の画像の組み合わせをインストールするか、 here とします。

以下の画像とリンクは、Linux、macOS、およびWindows上でのCUDAとTensorFlowの正式にサポート/テストされた組み合わせの概要を示しています。

マイナー構成

以下の仕様は広すぎる可能性があるため、ここでは機能する特定の構成を1つ示します。

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

対応するcudnnはダウンロードすることができます ここ 。

（2018年2月16日更新の数値）

Linux GPU

enter image description here

Linux

enter image description here

macOS GPU

enter image description here

マックOS

enter image description here

（2018年5月31日更新の図）

Windows

enter image description here

80
Fábio

https://www.tensorflow.org/install/source#tested_build_configurations にある互換性テーブルには、cudaとcuDNNの特定のマイナーバージョンは含まれていません。それは一般的にcuda = 9とcuDNN = 7としてリストされているだけです。ただし、特定のバージョンが一致しないと、エラーが発生します。

tensorflow-gpu==1.12.0とcuda==9.0の場合、互換性のあるcuDNNのバージョンは7.1.4です。これは登録後に here からダウンロードできます。

あなたはcudaのバージョンをチェックすることができます。
nvcc --version

cuDNNのバージョン
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2

tensorflow-gpuバージョン
pip freeze | grep tensorflow-gpu

11
Atul Balaji

動作：tensorflow 1.13.1、CUDA 10、CUDNN 7.4.2、python 3.6（3.7ではうまく動作しません。3.7には多くのバグがあります）

3
Ketki Shroff

あなたはcuda 10.0（10.1は3/18の時点では動作しません）にこの設定を使うことができます、これは私のために走ります：

  • テンソルフロー> = 1.12.0
  • tensorflow_gpu> = 1.4

バージョンtensorflow gpuをインストールしてください。

pip install tensorflow-gpu==1.4.0
2
ED-DOUGHMI younes