Процесс обучения ML не на GPU

Я только что перешел с AWS на Google Cloud Platform из-за более низкой стоимости графического процессора. Я выполнил инструкции на веб-сайте, создав экземпляр вычислительного движка с графическим процессором K80, установил последнюю версию Tensorflow, keras, драйвер cuda и cnDNN, все идет очень хорошо. Однако, когда я пытаюсь обучить свою модель, процесс обучения все еще выполняется на CPU.

NVIDIA-SMI 387.26                 Driver Version: 387.26        
Cuda compilation tools, release 9.1, V9.1.85
Tensorflow version -1.4.1
cudnn cudnn-9.1-linux-x64-v7
Ubuntu 16.04

person Can Cui    schedule 23.12.2017    source источник


Ответы (1)


Возможно, вы установили тензорную версию процессора?

Теперь вычислительный движок Google Cloud имеет образ ОС виртуальной машины со всем необходимым программным обеспечением, установленным для более простого / быстрого начала работы: https://cloud.google.com/deep-learning-vm/

person Chris Kleban    schedule 11.12.2018