18 lines
502 B
Python
18 lines
502 B
Python
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()
|
|
|