Я использую виртуальную машину 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
]