代码收藏家 TensorFlow和CUDA、cudnn、Pytorch以及英伟达显卡对应版本对照表
TensorFlow和CUDA、cudnn、Pytorch以及英伟达显卡对应版本对照表
一、TensorFlow对应版本对照表
| 版本 | Python 版本 | 编译器 | cuDNN | CUDA |
|---|---|---|---|---|
| tensorflow-2.9.0 | 3.7-3.10 | 8.1 | 11.2 | |
| tensorflow-2.8.0 | 3.7-3.10 | 8.1 | 11.2 | |
| tensorflow-2.7.0 | 3.7-3.9 | 8.1 | 11.2 | |
| tensorflow-2.6.0 | 3.6-3.9 | GCC 7.3.1 | 8.1 | 11.2 |
| tensorflow-2.5.0 | 3.6-3.9 | GCC 7.3.1 | 8.1 | 11.2 |
| tensorflow-2.4.0 | 3.6-3.8 | GCC 7.3.1 | 8.0 | 11.0 |
| tensorflow-2.3.0 | 3.5-3.8 | GCC 7.3.1 | 7.6 | 10.1 |
| tensorflow-2.2.0 | 3.5-3.8 | GCC 7.3.1 | 7.6 | 10.1 |
| tensorflow-2.1.0 | 2.7、3.5-3.7 | GCC 7.3.1 | 7.6 | 10.1 |
| tensorflow-2.0.0 | 2.7、3.3-3.7 | GCC 7.3.1 | 7.4 | 10.0 |
| tensorflow_gpu-1.15.0 | 2.7、3.3-3.7 | GCC 7.3.1 | 7.4 | 10.0 |
| tensorflow_gpu-1.14.0 | 2.7、3.3-3.7 | GCC 4.8 | 7.4 | 10.0 |
| tensorflow_gpu-1.13.1 | 2.7、3.3-3.7 | GCC 4.8 | 7.4 | 10.0 |
| tensorflow_gpu-1.12.0 | 2.7、3.3-3.6 | GCC 4.8 | 7 | 9 |
| tensorflow_gpu-1.11.0 | 2.7、3.3-3.6 | GCC 4.8 | 7 | 9 |
| tensorflow_gpu-1.10.0 | 2.7、3.3-3.6 | GCC 4.8 | 7 | 9 |
| tensorflow_gpu-1.9.0 | 2.7、3.3-3.6 | GCC 4.8 | 7 | 9 |
| tensorflow_gpu-1.8.0 | 2.7、3.3-3.6 | GCC 4.8 | 7 | 9 |
| tensorflow_gpu-1.7.0 | 2.7、3.3-3.6 | GCC 4.8 | 7 | 9 |
| tensorflow_gpu-1.6.0 | 2.7、3.3-3.6 | GCC 4.8 | 7 | 9 |
| tensorflow_gpu-1.5.0 | 2.7、3.3-3.6 | GCC 4.8 | 7 | 9 |
| tensorflow_gpu-1.4.0 | 2.7、3.3-3.6 | GCC 4.8 | 6 | 8 |
| tensorflow_gpu-1.3.0 | 2.7、3.3-3.6 | GCC 4.8 | 6 | 8 |
| tensorflow_gpu-1.2.0 | 2.7、3.3-3.6 | GCC 4.8 | 5.1 | 8 |
| tensorflow_gpu-1.1.0 | 2.7、3.3-3.6 | GCC 4.8 | 5.1 | 8 |
| tensorflow_gpu-1.0.0 | 2.7、3.3-3.6 | GCC 4.8 | 5.1 | 8 |
二、Pytorch对应版本对照表
| PyTorch 版本 | CUDA 环境 |
|---|---|
| 0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1) | 9.2 |
| 1.2.0、1.1.0、1.0.0(1) | 10.0 |
| 1.4.0、1.5.0(1)、1.6.0、1.7.0(1) | 10.1 |
| 1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.10.0 | 10.2 |
| 1.7.0(1) | 11.0 |
| 1.8.0(1)、1.9.0、1.10.0 | 11.1 |
| 1.8.0(1)、1.9.0、1.10.0 | 11.3 |
三、英伟达显卡算力
| CUDA Toolkit | Toolkit Driver Version Linux() |
|---|---|
| CUDA 11.7 Update 1 | >=515.65.01 |
| CUDA 11.7 GA | >=515.43.04 |
| CUDA 11.6 Update 2 | >=510.47.03 |
| CUDA 11.6 Update 1 | >=510.47.03 |
| CUDA 11.6 GA | >=510.39.01 |
| CUDA 11.5 Update 2 | >=495.29.05 |
| CUDA 11.5 Update 1 | >=495.29.05 |
| CUDA 11.5 GA | >=495.29.05 |
| CUDA 11.4 Update 4 | >=470.82.01 |
| CUDA 11.4 Update 3 | >=470.82.01 |
| CUDA 11.4 Update 2 | >=470.57.02 |
| CUDA 11.4 Update 1 | >=470.57.02 |
| CUDA 11.4.0 GA | >=470.42.01 |
| CUDA 11.3.1 Update 1 | >=465.19.01 |
| CUDA 11.3.0 GA | >=465.19.01 |
| CUDA 11.2.2 Update 2 | >=460.32.03 |
| CUDA 11.2.1 Update 1 | >=460.32.03 |
| CUDA 11.2.0 GA | >=460.27.03 |
| CUDA 11.1.1 Update 1 | >=455.32 |
| CUDA 11.1 GA | >=455.23 |
| CUDA 11.0.3 Update 1 | >= 450.51.06 |
| CUDA 11.0.2 GA | >= 450.51.05 |
| CUDA 11.0.1 RC | >= 450.36.06 |
| CUDA 10.2.89 | >= 440.33 |
| CUDA 10.1 (10.1.105 general release, and updates) | >= 418.39 |
| CUDA 10.0.130 | >= 410.48 |
| CUDA 9.2 (9.2.148 Update 1) | >= 396.37 |
| CUDA 9.2 (9.2.88) | >= 396.26 |
| CUDA 9.1 (9.1.85) | >= 390.46 |
| CUDA 9.0 (9.0.76) | >= 384.81 |
| CUDA 8.0 (8.0.61 GA2) | >= 375.26 |
| CUDA 8.0 (8.0.44) | >= 367.48 |
| CUDA 7.5 (7.5.16) | >= 352.31 |
| CUDA 7.0 (7.0.28) | >= 346.46 |
