mirror of https://github.com/nicolargo/glances.git
add comparable NVML_LIB check for Windows
This commit is contained in:
parent
f12de6619b
commit
30a31eda81
|
|
@ -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 as ose:
|
||||
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
|
||||
|
|
|
|||
Loading…
Reference in New Issue