import subprocess

command = ['nvidia-smi', '--query-gpu=name,memory.total,memory.used,driver_version',
           '--format=csv']
try:
    result = subprocess.run(command, capture_output=True, text=True, timeout=15,
                            check=True)
    print(result.stdout)
except FileNotFoundError:
    print('Brak nvidia-smi. Ten skrypt sprawdza karty Nvidia ze sterownikiem.')
except (subprocess.CalledProcessError, subprocess.TimeoutExpired) as error:
    print('Nie udalo sie odczytac karty:', error)
