J'essaie d'utiliser tensorflow sur ma machine Ubuntu 16.04, mais même après l'installation de tensorflow gpu; Je ne peux pas utiliser gpu pour mon tensorflow

Question

J'essaie d'utiliser le GPU tensorflow sur ma machine Ubuntu 16.04. J'ai réussi à installer CUDA toolkit (8.0.61) et cuDNN (6.0.21). Le problème est, je ne peux pas utiliser gpu tensorflow même après ce processus d'installation. Alors que l'importation de tensorflow ne montre aucune ligne d'importation pour le GPU tensorflow. its not showing any lines while importing tensorflow, tandis que mon autre machine ubuntu montre d'autres lignes lors de l'importation de tensorflow.

Nvidia driver version and usage

Answer 1

Il n'y a pas de lignes lors de l'importation tensorflow. Seulement à partir d'une session vous donne une sortie:

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

Après avoir consulté ma boule de cristal, je vous suggère de vérifier plus la variable d'environnement TF_CPP_MIN_LOG_LEVEL.