d5c810f 0e61e5c
1
2
3
4
5
6
import torch import streamlit as st viz = torch.cuda.memory_summary() st.write(viz)