GPU не может обнаружить tenorflow-GPU 2.0 - PullRequest
0 голосов
/ 27 октября 2019

GPU не может определить tenorflow-gpu 2.0, в чем проблема? для

import tensorflow as tf
print(tf.test.is_built_with_cuda())
print(tf.test.is_gpu_available())
from tensorflow.python.client import device_lib

device_lib.list_local_devices()

вывод на моем компьютере ниже с Ubuntu 16.04

True
False
[name: "/device:CPU:0"
 device_type: "CPU"
 memory_limit: 268435456
 locality {
 }
 incarnation: 10185800011826802770, name: "/device:XLA_CPU:0"
 device_type: "XLA_CPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 2908743746130792497
 physical_device_desc: "device: XLA_CPU device", name: "/device:XLA_GPU:0"
 device_type: "XLA_GPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 13796634005997726084
 physical_device_desc: "device: XLA_GPU device", name: "/device:XLA_GPU:1"
 device_type: "XLA_GPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 4940094218016527544
 physical_device_desc: "device: XLA_GPU device", name: "/device:XLA_GPU:2"
 device_type: "XLA_GPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 15391692776259568109
 physical_device_desc: "device: XLA_GPU device", name: "/device:XLA_GPU:3"
 device_type: "XLA_GPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 8368387340417357161
 physical_device_desc: "device: XLA_GPU device"]

1 Ответ

0 голосов
/ 28 октября 2019

Как указал y.selivonchyk, теперь он работает после установки cuda 10.0. Хотя туда нелегко (включая удаление / переустановку драйвера nvidia и cuda), если кому-то понадобится помощь, дайте мне знать в комментариях.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...