# USE THIS FILE TO CLEAR GPU CACHE import gc import torch # def report_gpu(): print(torch.cuda.list_gpu_processes()) gc.collect() torch.cuda.empty_cache()