mapleadmin/GPT

1import streamlit as st2import subprocess3 4def main():5    st.title("GPU Utilization Monitor")6 7    # Run nvidia-smi and capture the output8    try:9        nvidia_smi_output = subprocess.check_output(["nvidia-smi", "--query-gpu=utilization.gpu,utilization.memory", "--format=csv,noheader,nounits"])10        gpu_utilization = nvidia_smi_output.decode("utf-8").strip().split("\n")11    except subprocess.CalledProcessError as e:12        st.error(f"Error running nvidia-smi: {e}")13        return14 15    if not gpu_utilization:16        st.error("No GPU utilization data found.")17        return18 19    selected_gpu = 0  # Assuming you want to monitor the first GPU20 21    # Parse GPU utilization information22    gpu_info = gpu_utilization[selected_gpu].split(",")23    gpu_utilization_percentage = float(gpu_info[0])24    memory_utilization_percentage = float(gpu_info[1])25 26    st.subheader(f"GPU {selected_gpu} Utilization:")27    st.text(f"GPU Utilization: {gpu_utilization_percentage}%")28    st.text(f"Memory Utilization: {memory_utilization_percentage}%")29 30if __name__ == "__main__":31    main()32