TensorFlow从1.2到最新版别的CUDA和cuDNN所对应的版别调集。

TensorFlow版别 CUDA版别 cuDNN版别
1.2 CUDA Toolkit 8.0 cuDNN v5.1
1.3 CUDA Toolkit 8.0 cuDNN v6 or v6.1
1.4 CUDA Toolkit 8.0 cuDNN v6.1
1.5 CUDA Toolkit 9.0 cuDNN v7.0
1.6 CUDA Toolkit 9.0 cuDNN v7.0
1.7 CUDA Toolkit 9.0 cuDNN v7.0
1.8 CUDA Toolkit 9.0 cuDNN v7.0
1.9 CUDA Toolkit 9.0 cuDNN v7.0
1.10 CUDA Toolkit 9.0 cuDNN v7.1
1.11 CUDA Toolkit 9.0 cuDNN v7.1
1.12 CUDA Toolkit 9.0 cuDNN v7.3
1.13 CUDA Toolkit 10.0 cuDNN v7.3
1.14 CUDA Toolkit 10.0 cuDNN v7.4
1.15 CUDA Toolkit 10.0 cuDNN v7.5
2.0 CUDA Toolkit 10.0 cuDNN v7.6
2.1.0 CUDA Toolkit 10.1 cuDNN v7.6