torch.cuda.empty_cache() defaults to cuda:0 device unless explicitly set...
torch.cuda.empty_cache() defaults to cuda:0 device unless explicitly set otherwise first. Updating torch_gc() to use the device set by --device-id if specified to avoid OOM edge cases on multi-GPU systems.
Showing
Please
register
or
sign in
to comment