Amazing, it works great!
I was using the nvidia-smi
to determine the memory allocation on the GPU. What would be the difference between that and torch.cuda.memory_allocated
.
Amazing, it works great!
I was using the nvidia-smi
to determine the memory allocation on the GPU. What would be the difference between that and torch.cuda.memory_allocated
.