代码之家 › 专栏 › 技术社区 › whiletrue

哪些tensorflow和cuda版本的组合是兼容的?

cudnn compatibility version cuda tensorflow

64

whiletrue · 技术社区 · 8 年前

我注意到一些较新的tensorflow版本与较旧的cuda和cudnn版本不兼容。是否有兼容版本的概述,甚至官方测试的组合列表?我在TensorFlow文档中找不到。

4 回复 | 直到 8 年前

1

107

whiletrue 7 年前

一般情况下:

检查CUDA版本:

cat /usr/local/cuda/version.txt

和CUDNN版本:

grep CUDNN_MAJOR -A 2 /usr/local/cuda/include/cudnn.h

并按下图所示安装组合,或 here .

以下图片和链接概述了Linux、MacOS和Windows上正式支持/测试的CUDA和TensorFlow组合:

由于在某些情况下下面给出的规范可能过于宽泛,下面是一个可以工作的特定配置:

相应的CUDNN可以下载 here .

(数据于2019年6月29日更新)

(2018年5月31日更新)

2

12

Atul Balaji 7 年前

中给出的兼容性表 https://www.tensorflow.org/install/source#tested_build_configurations 不包含CUDA和CUDNN的特定次要版本。一般只列为cuda=9和cudnn=7。但是,如果不满足特定版本,则会出现错误。

为了 tensorflow-gpu==1.12.0 和 cuda==9.0 ,兼容的 cuDNN 版本是 7.1.4 ,可从 here 注册后。

您可以使用
nvcc --version

CUDNN版本使用
cat /usr/include/cudnn.h | grep CUDNN_MAJOR -A 2

TensorFlow GPU版本使用
pip freeze | grep tensorflow-gpu

3

4

Ketki Shroff 7 年前

工作方式:TensorFlow 1.13.1、CUDA 10、CUDNN 7.4.2、Python 3.6(与3.7不太配合)。3.7有很多错误) 对于Windows 10

4

1

Evan Mata 7 年前

您可以将此配置用于CUDA 10.0(10.1自3/18起不工作),这对我运行:

安装TensorFlow GPU版本:

pip install tensorflow-gpu==1.4.0