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 |
