TensorFlow官网安装指导界面提示“运行支持 GPU 的 TensorFlow 所需满足的 NVIDIA 要求”
查看机器CUDA、cuDNN版本命令分别是:
- cat /usr/local/cuda/version.txt
- 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为例:
可以看到1.0~1.2是我们可以使用的GPU版。