import paramiko

c = paramiko.SSHClient()
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
c.connect('140.44.4.90', username='rancher', key_filename=r'C:\Users\Snarf\.ssh\id_ed25519_snarfnet', timeout=10)

def run(cmd):
    stdin, stdout, stderr = c.exec_command(f"echo Katzir476! | sudo -S sh -c '{cmd}'")
    return stdout.read().decode().strip()

print("Check nvidia-smi:")
print(run("which nvidia-smi || nvidia-smi"))
print("Check GPU devices:")
print(run("ls -l /dev/nvidia*"))

c.close()