Ubuntu下配置TensorFlow-Gpu版,其对应CUDA、cuDNN版本

TensorFlow官网安装指导界面提示“运行支持 GPU 的 TensorFlow 所需满足的 NVIDIA 要求”

查看机器CUDA、cuDNN版本命令分别是:

  1. cat /usr/local/cuda/version.txt
  2. cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 例如得到如下信息:

CUDA Version 8.0.44

#define CUDNN_MAJOR      5
#define CUDNN_MINOR      1
#define CUDNN_PATCHLEVEL 5

 这表示CUDA为8.0版,cuDNN为5.1.5

官网给出了经过测试的源代码配置信息:

以linux为例:

Ubuntu下配置TensorFlow-Gpu版,其对应CUDA、cuDNN版本

Ubuntu下配置TensorFlow-Gpu版,其对应CUDA、cuDNN版本

 可以看到1.0~1.2是我们可以使用的GPU版。