安装Tensorflow-gpu后,我收到错误消息无法加载本机TensorFlow运行时

时间:2019-05-27 11:38:49

标签: tensorflow

我已经安装了redhat 6.6,cuda 9.0,cudnn 7.3.1和tesnorflow-gpu。但是安装后出现以下错误。我的显卡是nvidia titan v 追溯(最近一次通话):

<Label Text="Sample &amp;Text"/>

1 个答案:

答案 0 :(得分:1)

您是否测试了cuda和cudnn的安装:

Nvidia驱动程序:

$ nvidia-smi

这应该向您显示GPU指标的概述

测试CUDA:

请先检查是否:

$ nvcc -V

显示正确的cuda工具包版本

然后您可以通过以下过程对其进行测试: 首先:

 $ cd ~/NVIDIA_CUDA-9.0_Samples
 $ make

然后:

$ cd ~/NVIDIA_CUDA-10.0_Samples/bin/x86_64/linux/release
$./deviceQuery

如果最后有一个“结果:通过”,那就一切都很好!

要测试cudnn:

$ cp -r /usr/src/cudnn_samples_v7/ $HOME
$ cd $HOME/cudnn_samples_v7/mnistCUDNN
$ make clean && make
$ ./mnistCUDNN

结果应该是:“测试通过!”

tensorflow:

如果cuda和cudnn在工作,我建议您使用以下命令在conda环境中安装tensorflow:

conda create --name tf_gpu tensorflow-gpu

对我来说(并且在遇到很多问题之后),它运行得很好。

要测试:

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

主要来源: install tensorflow GPU tensorflow for GPU made easy