This commit is contained in:
Deniz Şafak
2025-07-11 11:51:41 +03:00
parent c9450d2ca5
commit 7a8df9b34e
14 changed files with 881 additions and 406 deletions
+4 -2
View File
@@ -1,20 +1,22 @@
import gpustat
def check():
try:
stats = gpustat.new_query()
except Exception:
return False
nvidia_keywords = ['nvidia', 'rtx', 'gtx', 'quadro', 'tesla', 'titan', 'mx']
nvidia_keywords = ["nvidia", "rtx", "gtx", "quadro", "tesla", "titan", "mx"]
for gpu in stats.gpus:
name = gpu.name.lower()
if any(keyword in name for keyword in nvidia_keywords):
return True
return False
if __name__ == "__main__":
stats = gpustat.new_query()
for gpu in stats.gpus:
print(gpu.name)
print(check())
print(check())