mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-18 05:40:26 +02:00
23 lines
495 B
Python
23 lines
495 B
Python
import gpustat
|
|
|
|
|
|
def check():
|
|
try:
|
|
stats = gpustat.new_query()
|
|
except Exception:
|
|
return False
|
|
|
|
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())
|