我正在尝试在我的 ubuntu 16.04 机器上使用 tensorflow,但即使在安装了 tensorflow GPU 之后;我不能将 GPU 用于我的张量流



我正在尝试在我的Ubuntu 16.04机器上使用Tensorflow GPU。我已经成功安装了CUDA工具包(8.0.61)和Cudnn(6.0.21)。问题是,即使在此安装过程之后,我也无法使用TensorFlow GPU。

导入TensorFlow在导入TensorFlow时未显示任何导入行。

nvidia驱动程序版本和用法

导入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

最新更新