Почему тензорный поток на виртуальной машине глубокого обучения google не использует графический процессор?

Я использую виртуальную машину Google Deep Learning из Google Marketplace, и я выбрал графический процессор NvdiaK80. Я пытаюсь обучить модель обнаружения объектов с помощью API обнаружения объектов. Однако я заметил, что тензорный поток по умолчанию не использует графический процессор (код для проверки ниже)

Я предполагаю, что этот экземпляр поставляется со всеми необходимыми драйверами NVIDIA, поэтому это не проблема, связанная с драйверами.

Дальнейшее расследование показало, что у меня было 2 установки Tensorflow (tensorflow 1.12.0 и tensorflow-GPU 1.12.0). Поэтому я удалил версию процессора. Однако это все равно не помогает.

Я использовал приведенный ниже код, чтобы проверить, использует ли тензорный поток графический процессор

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

Для справки я использую приведенный ниже код для обучения обнаружению объектов, который отлично работает на виртуальной машине с глубоким обучением, но не использует графический процессор.

python $Tensor_path/legacy/train.py --logtostderr -- 
train_dir=$Train_path/training/ -- 
pipeline_config_path=$Train_path/training/
ssd_inception_v2_pets.config

Выход (я ожидал, что особенности используемого устройства GPU)

[name: "/cpu:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 18292259467280600161
]

person sourav    schedule 22.12.2018    source источник


Ответы (1)


Мне удалось решить эту проблему, удалив старый экземпляр и начав заново с нового. Я предполагаю, что установка графического процессора тензорного потока была повреждена при установке API обнаружения объектов. Выполнив указанные здесь шаги, установите https://cloud.google.com/solutions/creating-object-detection-application-tensorflow

И скорее всего виновата именно эта строчка

pip install --upgrade 
https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.1.0-cp27-none- 
linux_x86_64.whl
person sourav    schedule 24.12.2018