Как узнать, что программа выполняется на GPU или CPU на Ubuntu?

Question

Я использую Ubuntu 16.04 и запускаю программу для обучения модели DeepLearning.Число эпох кажется большим, но программа работает слишком медленно.Я хочу убедиться, что моя программа работает на GPU?

Откуда мне это знать?

Спасибо!

Answer 1

Я предполагаю, что вы используете графический процессор Nvidia.

Самый простой способ добиться этого - набрать следующую команду внутри терминала:

watch -n 1 nvidia-smi

Это даст вам непрерывное обновление (каждую секунду) без заполнения терминала выводом.

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