dgbpy.keras_fix
Module Contents
Functions
Get a list of available gpu devices (formatted as strings). |
Attributes
Get a list of available gpu devices (formatted as strings). |
- dgbpy.keras_fix._get_available_gpus()
Get a list of available gpu devices (formatted as strings).
Workaround for TensorFlow 2 error when loading TensorFlow 1 models in Keras
- # Returns
A list of available GPU devices.
- dgbpy.keras_fix._get_available_gpus