GPU usage by NEMO ASR Model
You can try using this before important anything from NeMo - https://github.com/pytorch/pytorch/blob/main/torch/cuda/memory.py#L121
I want to limit the usage of Nemo[Asr] mode, lets say i want to limit it to 5 GB of VRAM. how can i do it? is there any way to limit this?