无法在GPU上运行tensorflow

Question

我无法按照安装指南（https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/get_started/os_setup.md#installing-from-sources）中的说明运行TF-CUDA tutorials_example_trainer

我以前遇到过CUDA库的问题，但那是与图形相关的演示版。

以下所有详细信息， 提前感谢您提供的帮助。

环境信息

操作系统：Debian Stretch

已安装的CUDA和cuDNN版本： 8.0,5.0

如果从源代码安装，请提供

1. 554ddd9ad2d4abad5a9a31f2d245f0b1012f0d10
2. 构建标签：0.3.0 构建目标：bazel-out / local-fastbuild / bin / src / main / java / com / google / devtools / build / lib / bazel / BazelServer_deploy.jar 建造时间：2016年6月10日星期五11:38:23（1465558703）

重现的步骤

1. 使用367.35驱动程序从源代码构建
2. 运行std::string

有用的日志或其他输出

bazel-bin/tensorflow/cc/tutorials_example_trainer --use_gpu

Answer 1

错误消息表明您的GPU驱动程序设置不正确。您可以尝试以下命令来查看驱动程序是否已正确安装。

$ nvidia-smi

如果没有，请按照CUDA官方网站上的说明重新安装CUDA。由于您的操作系统未得到官方支持，您可能需要更改操作系统。