Tensorflow与CUDA、cudnn版本对应关系

不同版本的Tensorflow需对应不同的CUDA和cudnn版本,否者容易安装失败。可按下图所示,根据想要安装的Tensorflow版本,选择对应版本的CUDA和cudnn。

其中CUDA的下载链接为:

CUDA Toolkit Archive | NVIDIA Developer

cudnn下载链接(需要登录开发者账号)为:

cuDNN Archive | NVIDIA Developer

GPU

Version Python version cuDNN CUDA
tensorflow_gpu-2.11.0 3.7-3.10 8.1 11.2
tensorflow_gpu-2.10.0 3.7-3.10 8.1 11.2
tensorflow_gpu-2.9.0 3.7-3.10 8.1 11.2
tensorflow_gpu-2.8.0 3.7-3.10 8.1 11.2
tensorflow_gpu-2.7.0 3.7-3.9 8.1 11.2
tensorflow_gpu-2.6.0 3.6-3.9 8.1 11.2
tensorflow_gpu-2.5.0 3.6-3.9 8.1 11.2
tensorflow_gpu-2.4.0 3.6-3.8 8.0 11.0
tensorflow_gpu-2.3.0 3.5-3.8 7.6 10.1
tensorflow_gpu-2.2.0 3.5-3.8 7.6 10.1
tensorflow_gpu-2.1.0 3.5-3.7 7.6 10.1
tensorflow_gpu-2.0.0 3.5-3.7 7.4 10
tensorflow_gpu-1.15.0 3.5-3.7 7.4 10
tensorflow_gpu-1.14.0 3.5-3.7 7.4 10
tensorflow_gpu-1.13.0 3.5-3.7 7.4 10
tensorflow_gpu-1.12.0 3.5-3.6 7.2 9.0
tensorflow_gpu-1.11.0 3.5-3.6 7 9
tensorflow_gpu-1.10.0 3.5-3.6 7 9
tensorflow_gpu-1.9.0 3.5-3.6 7 9
tensorflow_gpu-1.8.0 3.5-3.6 7 9
tensorflow_gpu-1.7.0 3.5-3.6 7 9
tensorflow_gpu-1.6.0 3.5-3.6 7 9
tensorflow_gpu-1.5.0 3.5-3.6 7 9
tensorflow_gpu-1.4.0 3.5-3.6 6 8
tensorflow_gpu-1.3.0 3.5-3.6 6 8
tensorflow_gpu-1.2.0 3.5-3.6 5.1 8
tensorflow_gpu-1.1.0 3.5 5.1 8
tensorflow_gpu-1.0.0 3.5 5.1 8

猜你喜欢

转载自blog.csdn.net/chan1987818/article/details/126264482