Факел обнаруживает GPU, но TF не может его найти - PullRequest
Новая
       11

Факел обнаруживает GPU, но TF не может его найти

0 голосов
/ 25 мая 2019

Я выполнил все инструкции, и теперь факел работает на GPU, но TF не может обнаружить GPU.

Я перепробовал множество тем StackOverflow, но в конце концов я не смог!

результат этого:

torch.cuda.is_available()

равен:

True

Но когда я бегу: 

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

результат: 

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 6820542267928817272
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 16263925709367684790
physical_device_desc: "device: XLA_CPU device"
]

, а результат

pip3 list

: 

tensorboard                      1.13.1         
tensorflow                       1.13.1              
tensorflow-estimator             1.13.0              
tensorflow-estimator-2.0-preview 1.14.0.dev2019052100
tensorflow-gpu                   1.13.1

Как мне запуститьмоя сеть на GPU с использованием TF.keras?

...