Tensorflow-GPU版本需要的CUDA版本以及Cudnn的对应关系

最新推荐文章于 2024-04-01 12:40:12 发布

原创最新推荐文章于 2024-04-01 12:40:12 发布 · 3.0k 阅读

CC 4.0 BY-SA版权

14 篇文章

订阅专栏

本文详细介绍了在Ubuntu系统上配置深度学习环境的具体步骤，包括CUDA、cuDNN及TensorFlow-GPU版本的兼容性信息。提供了从Python版本到构建工具的全面指导，适合初学者快速搭建深度学习开发平台。

部署运行你感兴趣的模型镜像

具体环境配置可参考：配置深度学习环境ubuntu16.10 + cuda9.2 + cudnn 7 + tensorflow-gpu
查看cuda 版本信息cat /usr/local/cuda/version.txt
查看cudnn 版本信息cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
以下表格更新可参考官网here或者官网there

版本	Python 版本	编译器	构建工具	cuDNN	CUDA
tensorflow-2.1.0	2.7、3.5-3.7	GCC 7.3.1	Bazel 0.27.1	7.6 10.1
tensorflow-2.0.0	2.7、3.3-3.7	GCC 7.3.1	Bazel 0.26.1	7.4	10.0
tensorflow_gpu-1.14.0	2.7、3.3-3.7	GCC 4.8	Bazel 0.24.1	7.4	10.0
tensorflow_gpu-1.13.1	2.7、3.3-3.7	GCC 4.8	Bazel 0.19.2	7.4	10.0
tensorflow_gpu-1.12.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.15.0	7	9
tensorflow_gpu-1.11.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.15.0	7	9
tensorflow_gpu-1.10.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.15.0	7	9
tensorflow_gpu-1.9.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.11.0	7	9
tensorflow_gpu-1.8.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.10.0	7	9
tensorflow_gpu-1.7.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.9.0	7	9
tensorflow_gpu-1.6.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.9.0	7	9
tensorflow_gpu-1.5.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.8.0	7	9
tensorflow_gpu-1.4.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.5.4	6	8
tensorflow_gpu-1.3.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.4.5	6	8
tensorflow_gpu-1.2.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.4.5	5.1	8
tensorflow_gpu-1.1.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.4.2	5.1	8
tensorflow_gpu-1.0.0	2.7、3.3-3.6	GCC 4.8	Bazel 0.4.2	5.1	8