def get_system_specs()

in NVIDIA_runner.py [0:0]

• 18 lines of code
• 3 McCabe index (conditional complexity)

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()