Upload 6 files

README.md

@@ -1,12 +1,12 @@
 ---
 title: PLUS Lab GPUs
-emoji: 📉
-colorFrom: pink
-colorTo: blue
+emoji: ➕
+colorFrom: blue
+colorTo: purple
 sdk: gradio
-sdk_version: 5.14.0
+sdk_version: 5.13.1
 app_file: app.py
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,76 @@
+import gradio as gr
+import pandas as pd
+import plotly.express as px
+
+def split_multi_users(dfs):
+    df = dfs.copy()
+    df["usernames"] = df["username"].apply(lambda x: x.split(", "))
+    df["count"] = 1
+    new_df = []
+    for row in df.to_dict(orient="records"):
+        gpu_users_num = len(row["usernames"])
+        for username in row["usernames"]:
+            new_row = row.copy()
+            new_row["count"] = 1 / gpu_users_num
+            new_row["username"] = username
+            new_df.append(new_row)
+    df = pd.DataFrame(new_df)
+    return df
+
+def plot_now():
+    dfs = pd.read_csv("hf://spaces/mohsenfayyaz/PLUS_Lab_GPUs/gpus.csv")
+    dfs = dfs.drop(columns=["Unnamed: 0"])
+    dfs = dfs.fillna("FREE")
+    dfs_plot = split_multi_users(dfs)
+    fig = px.bar(
+        dfs_plot, x="count", y="server", color="username",
+        title=f"Last Updated {min(dfs['timestamp'])}",
+        color_discrete_map={
+            "FREE": "black",
+        },
+        text=dfs_plot['username'].astype(str) + "<br>" + dfs_plot['device'].astype(str),
+    )
+    fig.update_layout(
+        yaxis={'categoryorder': 'array', 'categoryarray': dfs_plot["server"].unique()[::-1]},
+        barcornerradius=50,
+    )
+    fig.update_traces(textposition='inside', insidetextanchor='middle')
+    return fig, dfs
+
+def plot_history():
+    dfh = pd.read_pickle("hf://spaces/mohsenfayyaz/PLUS_Lab_GPUs/history.pkl.gz", )
+    dfh = dfh.fillna("FREE")
+    dfh = split_multi_users(dfh)
+    dfh = dfh[["polling_timestamp", "username", "count"]]
+    dfh = dfh.groupby(["polling_timestamp", "username"]).sum()
+    dfh = dfh.reset_index()
+    dfh = dfh.sort_values(by=["polling_timestamp", "count"], ascending=False)
+    fig = px.area(dfh, x="polling_timestamp", y="count", color='username', color_discrete_map={"FREE": "black",}, markers=True)
+    return fig, dfh
+
+
+def plot_figs():
+    fig_now, dfn = plot_now()
+    try:
+        fig_history, dfh = plot_history()
+    except Exception as e:
+        print(e)
+        fig_history = None
+        dfh = None
+    return fig_now, dfn, fig_history
+
+demo = gr.Interface(
+    fn=plot_figs,
+    inputs = [],
+    outputs = [
+        gr.Plot(label="GPU Status", elem_classes="plotcss"),
+        gr.Dataframe(label="GPU Status Details"),
+        gr.Plot(label="History", elem_classes="plotcss"),
+    ],
+    live=True,
+    flagging_options=[],
+    css=".plotcss {max-width: 820px !important;}"
+)
+
+if __name__ == "__main__":
+    demo.launch(debug=True)

gpus.csv

@@ -0,0 +1,45 @@
+,timestamp,server,name,device,utilization.gpu [%],utilization.memory [%],username
+0,2025/01/30 15:52:03.165,pluslab-a100.cs.ucla.edu,NVIDIA A100-SXM4-40GB,0,100 %,8 %,fabricehc
+1,2025/01/30 15:52:03.189,pluslab-a100.cs.ucla.edu,NVIDIA A100-SXM4-40GB,1,0 %,0 %,fabricehc
+2,2025/01/30 15:52:03.210,pluslab-a100.cs.ucla.edu,NVIDIA A100-SXM4-40GB,2,0 %,0 %,fabricehc
+3,2025/01/30 15:52:03.226,pluslab-a100.cs.ucla.edu,NVIDIA A100-SXM4-40GB,3,0 %,0 %,fabricehc
+0,2025/01/30 15:52:04.328,pluslab01-a100.cs.ucla.edu,NVIDIA A100 80GB PCIe,0,100 %,32 %,lucasbandarkar
+1,2025/01/30 15:52:04.343,pluslab01-a100.cs.ucla.edu,NVIDIA A100 80GB PCIe,1,1 %,0 %,lucasbandarkar
+2,2025/01/30 15:52:04.355,pluslab01-a100.cs.ucla.edu,NVIDIA A100 80GB PCIe,2,92 %,30 %,tparekh
+3,2025/01/30 15:52:04.363,pluslab01-a100.cs.ucla.edu,NVIDIA A100 80GB PCIe,3,100 %,42 %,lucasbandarkar
+4,2025/01/30 15:52:04.372,pluslab01-a100.cs.ucla.edu,NVIDIA A100 80GB PCIe,4,99 %,44 %,hzhang19
+5,2025/01/30 15:52:04.382,pluslab01-a100.cs.ucla.edu,NVIDIA A100 80GB PCIe,5,99 %,46 %,hzhang19
+6,2025/01/30 15:52:04.390,pluslab01-a100.cs.ucla.edu,NVIDIA A100 80GB PCIe,6,99 %,37 %,hzhang19
+7,2025/01/30 15:52:04.397,pluslab01-a100.cs.ucla.edu,NVIDIA A100 80GB PCIe,7,99 %,34 %,hzhang19
+0,2025/01/30 15:52:05.536,pluslab01-h100.cs.ucla.edu,NVIDIA H100 80GB HBM3,0,89 %,39 %,sidilu
+1,2025/01/30 15:52:05.558,pluslab01-h100.cs.ucla.edu,NVIDIA H100 80GB HBM3,1,97 %,42 %,sidilu
+2,2025/01/30 15:52:05.574,pluslab01-h100.cs.ucla.edu,NVIDIA H100 80GB HBM3,2,98 %,39 %,sidilu
+3,2025/01/30 15:52:05.589,pluslab01-h100.cs.ucla.edu,NVIDIA H100 80GB HBM3,3,100 %,44 %,sidilu
+4,2025/01/30 15:52:05.605,pluslab01-h100.cs.ucla.edu,NVIDIA H100 80GB HBM3,4,100 %,44 %,sidilu
+5,2025/01/30 15:52:05.623,pluslab01-h100.cs.ucla.edu,NVIDIA H100 80GB HBM3,5,98 %,39 %,sidilu
+6,2025/01/30 15:52:05.639,pluslab01-h100.cs.ucla.edu,NVIDIA H100 80GB HBM3,6,98 %,38 %,sidilu
+7,2025/01/30 15:52:05.655,pluslab01-h100.cs.ucla.edu,NVIDIA H100 80GB HBM3,7,100 %,43 %,sidilu
+0,2025/01/30 15:52:07.066,pluslab-a6000.cs.ucla.edu,NVIDIA RTX A6000,0,12 %,3 %,fabricehc
+1,2025/01/30 15:52:07.072,pluslab-a6000.cs.ucla.edu,NVIDIA RTX A6000,1,0 %,0 %,
+2,2025/01/30 15:52:07.077,pluslab-a6000.cs.ucla.edu,NVIDIA RTX A6000,2,0 %,0 %,tparekh
+3,2025/01/30 15:52:07.082,pluslab-a6000.cs.ucla.edu,NVIDIA RTX A6000,3,87 %,40 %,tparekh
+4,2025/01/30 15:52:07.086,pluslab-a6000.cs.ucla.edu,NVIDIA RTX A6000,4,97 %,97 %,meithnav
+5,2025/01/30 15:52:07.092,pluslab-a6000.cs.ucla.edu,NVIDIA RTX A6000,5,70 %,38 %,"tparekh, meithnav"
+6,2025/01/30 15:52:07.096,pluslab-a6000.cs.ucla.edu,NVIDIA RTX A6000,6,100 %,86 %,"tparekh, meithnav"
+7,2025/01/30 15:52:07.101,pluslab-a6000.cs.ucla.edu,NVIDIA RTX A6000,7,96 %,96 %,"tparekh, meithnav"
+0,2025/01/30 15:52:08.238,pluslab01-a6000.cs.ucla.edu,NVIDIA RTX A6000,0,95 %,86 %,yunzhi
+1,2025/01/30 15:52:08.244,pluslab01-a6000.cs.ucla.edu,NVIDIA RTX A6000,1,83 %,43 %,tparekh
+2,2025/01/30 15:52:08.247,pluslab01-a6000.cs.ucla.edu,NVIDIA RTX A6000,2,80 %,46 %,tparekh
+3,2025/01/30 15:52:08.249,pluslab01-a6000.cs.ucla.edu,NVIDIA RTX A6000,3,0 %,0 %,tparekh
+4,2025/01/30 15:52:08.253,pluslab01-a6000.cs.ucla.edu,NVIDIA RTX A6000,4,0 %,0 %,
+5,2025/01/30 15:52:08.257,pluslab01-a6000.cs.ucla.edu,NVIDIA RTX A6000,5,94 %,96 %,yunzhi
+6,2025/01/30 15:52:08.261,pluslab01-a6000.cs.ucla.edu,NVIDIA RTX A6000,6,0 %,0 %,
+7,2025/01/30 15:52:08.265,pluslab01-a6000.cs.ucla.edu,NVIDIA RTX A6000,7,0 %,0 %,lucasbandarkar
+0,2025/01/30 15:52:09.990,pluslab02-a6000.cs.ucla.edu,NVIDIA RTX A6000,0,95 %,95 %,meithnav
+1,2025/01/30 15:52:10.013,pluslab02-a6000.cs.ucla.edu,NVIDIA RTX A6000,1,97 %,100 %,meithnav
+2,2025/01/30 15:52:10.030,pluslab02-a6000.cs.ucla.edu,NVIDIA RTX A6000,2,97 %,100 %,meithnav
+3,2025/01/30 15:52:10.048,pluslab02-a6000.cs.ucla.edu,NVIDIA RTX A6000,3,98 %,100 %,meithnav
+4,2025/01/30 15:52:10.070,pluslab02-a6000.cs.ucla.edu,NVIDIA RTX A6000,4,99 %,100 %,hzhang19
+5,2025/01/30 15:52:10.089,pluslab02-a6000.cs.ucla.edu,NVIDIA RTX A6000,5,99 %,71 %,"tparekh, hzhang19"
+6,2025/01/30 15:52:10.117,pluslab02-a6000.cs.ucla.edu,NVIDIA RTX A6000,6,25 %,9 %,"tparekh, hzhang19"
+7,2025/01/30 15:52:10.133,pluslab02-a6000.cs.ucla.edu,NVIDIA RTX A6000,7,71 %,39 %,tparekh

history.pkl.gz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51419de7243376d764d9934d8f886c6fa6cd0f240e32f5b7626da757a1da0b45
+size 213660

requirements.txt

@@ -0,0 +1 @@
+plotly