CalvinBackup/facefusion
1
0 Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-10-14 01:20:51 +02:00
Code Issues Packages Projects Releases Wiki Activity

scope the nvidia_ml stuff into module alias

Browse Source
This commit is contained in:
henryruhs committed 2026-05-11 16:36:23 +02:00
1 parent 0816bea6d3
commit 74a9df35a6
2 files changed
+15 -16
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

facefusion/system.py
+15 -15
View File
@@ -7,7 +7,7 @@ from typing import List
import psutil import psutil
from facefusion import state_manager from facefusion import state_manager
from facefusion.libraries.nvidia_ml import create_memory_configuration, create_static_library, create_utilization_configuration from facefusion.libraries import nvidia_ml as nvidia_ml_module
from facefusion.types import DiskMetrics, ExecutionProvider, GraphicDevice, MemoryMetrics, Metrics, NetworkMetrics, ProcessorMetrics from facefusion.types import DiskMetrics, ExecutionProvider, GraphicDevice, MemoryMetrics, Metrics, NetworkMetrics, ProcessorMetrics
@@ -33,36 +33,36 @@ def detect_graphic_devices(execution_providers : List[ExecutionProvider]) -> Lis
def detect_nvidia_graphic_devices() -> List[GraphicDevice]: def detect_nvidia_graphic_devices() -> List[GraphicDevice]:
nvidia_ml = create_static_library() nvidia_ml_library = nvidia_ml_module.create_static_library()
graphic_devices : List[GraphicDevice] = [] graphic_devices : List[GraphicDevice] = []
if nvidia_ml: if nvidia_ml_library:
nvidia_ml.nvmlInit_v2() nvidia_ml_library.nvmlInit_v2()
device_count = ctypes.c_uint() device_count = ctypes.c_uint()
nvidia_ml.nvmlDeviceGetCount_v2(ctypes.byref(device_count)) nvidia_ml_library.nvmlDeviceGetCount_v2(ctypes.byref(device_count))
driver_version = ctypes.create_string_buffer(80) driver_version = ctypes.create_string_buffer(80)
nvidia_ml.nvmlSystemGetDriverVersion(driver_version, 80) nvidia_ml_library.nvmlSystemGetDriverVersion(driver_version, 80)
cuda_version = ctypes.c_int() cuda_version = ctypes.c_int()
nvidia_ml.nvmlSystemGetCudaDriverVersion(ctypes.byref(cuda_version)) nvidia_ml_library.nvmlSystemGetCudaDriverVersion(ctypes.byref(cuda_version))
for device_id in range(device_count.value): for device_id in range(device_count.value):
device_handle = ctypes.c_void_p() device_handle = ctypes.c_void_p()
nvidia_ml.nvmlDeviceGetHandleByIndex_v2(device_id, ctypes.byref(device_handle)) nvidia_ml_library.nvmlDeviceGetHandleByIndex_v2(device_id, ctypes.byref(device_handle))
name = ctypes.create_string_buffer(96) name = ctypes.create_string_buffer(96)
nvidia_ml.nvmlDeviceGetName(device_handle, name, 96) nvidia_ml_library.nvmlDeviceGetName(device_handle, name, 96)
memory = create_memory_configuration() memory = nvidia_ml_module.create_memory_configuration()
nvidia_ml.nvmlDeviceGetMemoryInfo(device_handle, ctypes.byref(memory)) nvidia_ml_library.nvmlDeviceGetMemoryInfo(device_handle, ctypes.byref(memory))
temperature = ctypes.c_uint() temperature = ctypes.c_uint()
nvidia_ml.nvmlDeviceGetTemperature(device_handle, 0, ctypes.byref(temperature)) nvidia_ml_library.nvmlDeviceGetTemperature(device_handle, 0, ctypes.byref(temperature))
utilization = create_utilization_configuration() utilization = nvidia_ml_module.create_utilization_configuration()
nvidia_ml.nvmlDeviceGetUtilizationRates(device_handle, ctypes.byref(utilization)) nvidia_ml_library.nvmlDeviceGetUtilizationRates(device_handle, ctypes.byref(utilization))
graphic_devices.append( graphic_devices.append(
{ {
@@ -118,7 +118,7 @@ def detect_nvidia_graphic_devices() -> List[GraphicDevice]:
} }
}) })
nvidia_ml.nvmlShutdown() nvidia_ml_library.nvmlShutdown()
return graphic_devices return graphic_devices
requirements.txt
-1
View File
@@ -3,7 +3,6 @@ numpy==2.4.2
onnx==1.20.1 onnx==1.20.1
onnxruntime==1.24.1 onnxruntime==1.24.1
opencv-python==4.13.0.92 opencv-python==4.13.0.92
nvidia-ml-py==13.590.48
psutil==7.2.2 psutil==7.2.2
tqdm==4.67.3 tqdm==4.67.3
scipy==1.16.3 scipy==1.16.3