in NVIDIA_runner.py [0:0]
def get_system_specs():
file = open("Outputs/system_specs.txt", "w")
results = subprocess.run(["nvidia-smi", "--query-gpu=gpu_name,vbios_version,driver_version,memory.total", "--format=csv"], stdout=subprocess.PIPE,stderr=subprocess.PIPE)
output = results.stdout.decode('utf-8').split('\n')[1].split(",")
file.write("GPU name : "+ output[0]+"\n")
file.write("VBIOS : "+ output[1]+"\n")
file.write("driver version : "+ output[2]+"\n")
file.write("GPU memory capacity : "+ output[3]+"\n")
results = subprocess.run("nvcc --version | grep release", shell=True, stdout=subprocess.PIPE,stderr=subprocess.PIPE)
cuda_version = results.stdout.decode('utf-8').split(",")[1].strip().split(" ")[1]
file.write("CUDA version : "+cuda_version+"\n")
if output[0].strip() != "NVIDIA Graphics Device" or "GB200" in output[0]:
results = subprocess.run("lsb_release -a | grep Release", shell=True, stdout=subprocess.PIPE,stderr=subprocess.PIPE)
ubuntu = results.stdout.decode('utf-8').strip().split("\t")[1]
file.write("ubuntu version : "+ubuntu+"\n")
file.write("pytorch version : {torch.__version__}\n")
file.close()
return output[0].strip()