A few quick ways to find out memory usage on 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
A few quick ways to find out memory usage on 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