show服务器软硬件infos

# -*- coding: utf-8 -*-
import os, cv2, torch, torchvision, subprocess
from IPython import embed

def report_cpu_infos():
    print('*' * 96)
    cmd = '\
        cat /proc/cpuinfo | grep "cpu cores" | uniq; \
        cat /proc/cpuinfo | grep "physical id" | sort | uniq | wc -l; \
        cat /proc/cpuinfo | grep "model name" |uniq'
    subprocess.run(cmd, shell=True)

def report_gpu_infos():
    print('*' * 96)
    try:  
        output = subprocess.check_output("nvidia-smi", shell=True)  
        items = output.decode('utf-8').split('\n')  
        [print(item) for item in items]
    except Exception as e:  
        print(f"Error: {e}")  
        return None  

def report_pkg_infos():
    print('*' * 96)
    cmd = 'python --version'
    os.system(cmd)
    cmd = 'pip --version'
    os.system(cmd)
    print('cv2.__version__', cv2.__version__)
    print('torch.__version__', torch.__version__)
    print('torch.__numcard__', torch.cuda.device_count())
    print('torch.cuda.__version__', torch.version.cuda)
    print('torch.cuda.is_available', torch.cuda.is_available())
    print('torchvision.__version__', torchvision.__version__)


def report_server_infos():
    report_cpu_infos()
    report_gpu_infos()
    report_pkg_infos()


if __name__ == "__main__":
    
    report_server_infos()