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)
    print('cv2.__version__', cv2.__version__)
    print('torch.__version__', torch.__version__)
    print('torchvision.__version__', torchvision.__version__)
 
 
def report_server_infos():
    report_cpu_infos()
    report_gpu_infos()
    report_pkg_infos()
 
 
if __name__ == "__main__":
    
    report_server_infos()