mirror of
https://github.com/deepfakes/faceswap
synced 2025-06-07 10:43:27 -04:00
6a3b674bef
* Remove tensorflow_probability requirement * setup.py - fix progress bars * requirements.txt: Remove pre python 3.9 packages * update apple requirements.txt * update INSTALL.md * Remove python<3.9 code * setup.py - fix Windows Installer * typing: python3.9 compliant * Update pytest and readthedocs python versions * typing fixes * Python Version updates - Reduce max version to 3.10 - Default to 3.10 in installers - Remove incompatible 3.11 tests * Update dependencies * Downgrade imageio dep for Windows * typing: merge optional unions and fixes * Updates - min python version 3.10 - typing to python 3.10 spec - remove pre-tf2.10 code - Add conda tests * train: re-enable optimizer saving * Update dockerfiles * Update setup.py - Apple Conda deps to setup.py - Better Cuda + dependency handling * bugfix: Patch logging to prevent Autograph errors * Update dockerfiles * Setup.py - Setup.py - stdout to utf-8 * Add more OSes to github Actions * suppress mac-os end to end test
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
#!/usr/bin/env python3
|
|
""" Dummy functions for running faceswap on CPU. """
|
|
from ._base import _GPUStats
|
|
|
|
|
|
class CPUStats(_GPUStats):
|
|
""" Holds information and statistics about the CPU on the currently running system.
|
|
|
|
Notes
|
|
-----
|
|
The information held here is not useful, but _GPUStats is dynamically imported depending on
|
|
the backend used, so we need to make sure this class is available for Faceswap run on the CPU
|
|
Backend.
|
|
|
|
The base :class:`_GPUStats` handles the dummying in of information when no GPU is detected.
|
|
|
|
Parameters
|
|
----------
|
|
log: bool, optional
|
|
Whether the class should output information to the logger. There may be occasions where the
|
|
logger has not yet been set up when this class is queried. Attempting to log in these
|
|
instances will raise an error. If GPU stats are being queried prior to the logger being
|
|
available then this parameter should be set to ``False``. Otherwise set to ``True``.
|
|
Default: ``True``
|
|
"""
|
|
|
|
def _get_device_count(self) -> int:
|
|
""" Detect the number of GPUs attached to the system. Always returns zero for CPU
|
|
backends.
|
|
|
|
Returns
|
|
-------
|
|
int
|
|
The total number of GPUs connected to the PC
|
|
"""
|
|
retval = 0
|
|
self._log("debug", f"GPU Device count: {retval}")
|
|
return retval
|
|
|
|
def _get_handles(self) -> list:
|
|
""" Obtain the device handles for all connected GPUs.
|
|
|
|
Returns
|
|
-------
|
|
list
|
|
An empty list for CPU Backends
|
|
"""
|
|
handles: list = []
|
|
self._log("debug", f"GPU Handles found: {len(handles)}")
|
|
return handles
|
|
|
|
def _get_driver(self) -> str:
|
|
""" Obtain the driver version currently in use.
|
|
|
|
Returns
|
|
-------
|
|
str
|
|
An empty string for CPU backends
|
|
"""
|
|
driver = ""
|
|
self._log("debug", f"GPU Driver: {driver}")
|
|
return driver
|
|
|
|
def _get_device_names(self) -> list[str]:
|
|
""" Obtain the list of names of connected GPUs as identified in :attr:`_handles`.
|
|
|
|
Returns
|
|
-------
|
|
list
|
|
An empty list for CPU backends
|
|
"""
|
|
names: list[str] = []
|
|
self._log("debug", f"GPU Devices: {names}")
|
|
return names
|
|
|
|
def _get_vram(self) -> list[int]:
|
|
""" Obtain the RAM in Megabytes for the running system.
|
|
|
|
Returns
|
|
-------
|
|
list
|
|
An empty list for CPU backends
|
|
"""
|
|
vram: list[int] = []
|
|
self._log("debug", f"GPU VRAM: {vram}")
|
|
return vram
|
|
|
|
def _get_free_vram(self) -> list[int]:
|
|
""" Obtain the amount of RAM that is available, in Megabytes, for the running system.
|
|
|
|
Returns
|
|
-------
|
|
list
|
|
An empty list for CPU backends
|
|
"""
|
|
vram: list[int] = []
|
|
self._log("debug", f"GPU VRAM free: {vram}")
|
|
return vram
|