Statistics and setup for PlaidML on AMD devices.
This module must be kept separate from Keras, and be called prior to any Keras import, as the plaidML Keras backend is set from this module.
Handles the initialization of PlaidML and the returning of GPU information for connected cards from the PlaidML library.
- log_level (str, optional) – The requested Faceswap log level. Also dictates the level that PlaidML logging is set at. Default:”INFO”
- log (bool, optional) – Whether this class should output to the logger. If statistics are being accessed during a
crash, then the logger may not be available, so this gives the option to turn logging off
in those kinds of situations. Default:
List of device indices for active GPU devices.
The total number of GPU Devices discovered.
pladml._DeviceConfigobjects for GPUs that PlaidML has discovered.
The driver versions for each GPU device that PlaidML has discovered.
The name of each GPU device that PlaidML has discovered.
The VRAM of each GPU device that PlaidML has discovered.
Setup PlaidML for AMD Cards.
Sets the Keras backend to PlaidML, loads the plaidML backend and makes GPU Device information from PlaidML available to
- log_level (str) – Faceswap’s log level. Used for setting the log level inside PlaidML
- exclude_devices (list) – A list of integers of device IDs that should not be used by Faceswap