import torch
import streamlit as st

st.write(f"torch.cuda.memory_summary()")