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

196 阅读 0 评论 130 点赞

我是靠谱客的博主直率天空，这篇文章主要介绍Ubuntu下配置TensorFlow-Gpu版，其对应CUDA、cuDNN版本，现在分享给大家，希望可以做个参考。

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版。

以上就是直率天空最近收集整理的关于Ubuntu下配置TensorFlow-Gpu版，其对应CUDA、cuDNN版本的全部内容，更多相关Ubuntu下配置TensorFlow-Gpu版内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。