Как конвертировать XLA_GPU в GPU - PullRequest
Новая
       15

Как конвертировать XLA_GPU в GPU

0 голосов
/ 15 апреля 2020

Моя ОС - Ubuntu 18.04, а мой GPU - GTX850M. Я использую драйверы nvidia 430.50, CUDA 10.1, CuDNN 9.0 и tensorflow-gpu 1.14.0. Когда я пытаюсь получить доступные устройства в тензорном потоке с 

from tensorflow.python.client import device_lib

device_lib.list_local_devices()

, я получаю это 

[name: "/device:CPU:0"
 device_type: "CPU"
 memory_limit: 268435456
 locality {
 }
 incarnation: 2293723676390825589,
 name: "/device:XLA_GPU:0"
 device_type: "XLA_GPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 15287372432461854293
 physical_device_desc: "device: XLA_GPU device",
 name: "/device:XLA_CPU:0"
 device_type: "XLA_CPU"
 memory_limit: 17179869184
 locality {
 }
 incarnation: 10399216684927698454
 physical_device_desc: "device: XLA_CPU device"]

Я могу использовать XLA_GPU для базовых c приложений (т. Е. Производство с постоянным тензорным потоком) но я не могу тренировать нейронную сеть. Как я могу преобразовать XLA_GPU в графический процессор для обучения глубоких нейронных сетей?

...