gpu-mem-usage

A few quick ways to find out memory usage on GPU

multiple ways to figure out the memory usages of GPU

    (nvidia-smi dmon --help)

    torch.cuda.memory_allocated
    torch.cuda.memory_reserved

pip3 install -r requirements.txt
python3 gpumem.py

pip3 install -r requirements.txt
python3 memreport.py

Name		Name	Last commit message	Last commit date
Latest commit History 16 Commits
mnist		mnist
LICENSE		LICENSE
README.md		README.md
gpumem.py		gpumem.py
memreport.py		memreport.py
requirements.txt		requirements.txt