Tensorflow работает на GPU на домашнем ПК, но не на Google Colab

Question

Я пытаюсь обучить нейронную сеть в этом репо на Google Colab: https://github.com/carpedm20/DCGAN-tensorflow

Я запускаю этот код в Google Colab (с аппаратным ускорителем, установленным на GPU), и обучение составляет 40в разы медленнее, чем на моем домашнем ПК, у которого GTX 1050 гораздо слабее. Поэтому я предполагаю, что на colab он не работает на GPU.Вывод на Google Colab действительно указывает на то, что GPU был найден:

Найден GPU по адресу: / device: GPU: 0

Кто-нибудь знает, почему это так?

Answer 1

Я думаю, причина этих данных в том, что нужно время, чтобы перейти от драйвера к colab'у, и попробовать GPU на параллельной функции 100 раз в обоих случаях, для которых не нужны данные для запуска и сообщить мне результаты

в конце концовя нашел пример , который мог бы закончить первую эпоху за 39 секунд, когда я попробую, кажется, что colab тратит время на передачу данных для каждой моей фотографии, которую я делал, и брал в первые эпохи 332 каждый раз после первоначальных действийзапустить оба, он получит 39 секунд, и я получу 332 с

я надеюсь, что эта помощь