Я только что перешел с 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