File size: 81 Bytes
d5c810f
 
 
5034c15
1
2
3
4
5
import torch
import streamlit as st

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