We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 647bd34 commit ae42bf5Copy full SHA for ae42bf5
gpustat/nvml.py
@@ -118,7 +118,7 @@ def nvmlDeviceGetMemoryInfo(handle):
118
"""
119
M = pynvml_monkeypatch
120
121
- if M.has_memoryinfo_v2 is not None:
+ if M.has_memoryinfo_v2 is None:
122
try:
123
pynvml._nvmlGetFunctionPointer("nvmlDeviceGetMemoryInfo_v2")
124
M.has_memoryinfo_v2 = True
@@ -138,7 +138,8 @@ def nvmlDeviceGetMemoryInfo(handle):
138
"Your NVIDIA driver requires a compatible version of "
139
"pynvml (>= 11.510.69) installed to display the correct "
140
"memory usage information (See #141 for more details). "
141
- "Please try `pip install --upgrade pynvml`.")
+ "Please try `pip install --upgrade pynvml`.",
142
+ category=UserWarning)
143
memory = M.original_nvmlDeviceGetMemoryInfo(handle)
144
145
return memory
0 commit comments