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

84 阅读 0 评论 56 点赞

我是靠谱客的博主直率天空，最近开发中收集的这篇文章主要介绍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版。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

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