d5c810f 5034c15
1
2
3
4
5
import torch import streamlit as st st.markdown(f"torch.cuda.memory_summary()")