我试图在我的ubuntu 16.04机器上使用tensorflow,但即使安装了tensorflow gpu;我不能用gpu作为我的张量流

时间:2017-09-17 08:12:05

标签: tensorflow tensorflow-gpu

我试图在我的ubuntu 16.04机器上使用tensorflow GPU。我成功安装了CUDA工具包(8.0.61)和cuDNN(6.0.21)。问题是,即使在这个安装过程之后我也不能使用tensorflow gpu。

虽然导入tensorflow没有显示tensorflow GPU的任何导入行。its not showing any lines while importing tensorflow,而其他我的其他ubuntu机器在导入tensorflow时显示一些行。

Nvidia driver version and usage

1 个答案:

答案 0 :(得分:0)

导入TensorFlow时没有行。只有开始会话才会给你一些输出:

>>> import tensorflow as tf
>>> tf.Session()
2017-09-17 20:06:20.174697: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX
2017-09-17 20:06:20.343531: I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:892] successful NUMA node read from SysFS had negative value (-1), but there must be at least one NUMA node, so returning NUMA node zero
2017-09-17 20:06:20.344062: I tensorflow/core/common_runtime/gpu/gpu_device.cc:965] Found device 0 with properties: 
name: GeForce GTX 960 major: 5 minor: 2 memoryClockRate(GHz): 1.329
pciBusID: 0000:01:00.0
totalMemory: 3.94GiB freeMemory: 3.61GiB
2017-09-17 20:06:20.344084: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1055] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX 960, pci bus id: 0000:01:00.0, compute capability: 5.2)
<tensorflow.python.client.session.Session object at 0x7f387d89f210>

在咨询我的水晶球后,我建议你进一步检查环境变量TF_CPP_MIN_LOG_LEVEL