Merge pull request #3000 from shidetian/patch-1

add comparable NVML_LIB check for Windows
This commit is contained in:
Nicolas Hennion
2024-11-06 14:52:21 +01:00
committed by GitHub

View File

@@ -8,6 +8,9 @@
"""NVidia Extension unit for Glances' GPU plugin."""
import os
import sys
from glances.globals import nativestr
from glances.logger import logger
@@ -17,7 +20,13 @@ try:
# Avoid importing pynvml if NVML_LIB is not installed
from ctypes import CDLL
CDLL(NVML_LIB)
if (sys.platform[:3] == "win"):
try:
CDLL(os.path.join(os.getenv("WINDIR", "C:/Windows"), "System32/nvml.dll"))
except OSError:
CDLL(os.path.join(os.getenv("ProgramFiles", "C:/Program Files"), "NVIDIA Corporation/NVSMI/nvml.dll"))
else:
CDLL(NVML_LIB)
import pynvml
except OSError:
nvidia_gpu_enable = False