Использование CPU и GPU в Keras (Tensorflow 2.1)

Question

Я успешно настроил TensorFlow 2.1.0 с доступом к моему графическому процессору:

Если я использую Keras (from tensorflow import keras), чтобы соответствовать некоторым Модель Sequential (как в примере здесь ), по умолчанию будет использоваться для этого GPU или CPU? Есть ли какая-то команда, чтобы узнать, какая из них используется Keras, и могу ли я как-нибудь это настроить самостоятельно? Мне бы очень хотелось, чтобы какая-то очень базовая модель c Keras, обученная на GPU и CPU, имела лучшее представление о разнице в производительности.

Answer 1

Поскольку TensorFlow 2.1, пакеты GPU и CPU находятся в одном пакете, tensorflow, не так, как в предыдущих версиях, которые имели отдельные версии для CPU и GPU: tensorflow и tensorflow-gpu.

Вы можете проверить, чтобы почувствовать себя лучше:

   #Use only CPU
   import os 
   os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

Или вы можете сделать вашу видеокарту видимой для TensorFlow, либо разрешив настройки по умолчанию, как указано выше, либо принудительно запустив ее с помощью:

os.environ['CUDA_VISIBLE_DEVICES']= '0'

Обратите внимание, что в приведенной выше настройке, например, если у вас было 4 графических процессора, вы должны установить:

 os.environ['CUDA_VISIBLE_DEVICES']= '0,1,2,3'