Hello! I’m working on making a inspector which examines each tensor, or nn.Module’s gpu/cpu memory resource consumption. (since nvidia-smi only shows total consumption) Is there any built-in pytorch method to achieve this goal? Thanks!

Inspect memory usage

smth September 30, 2017, 10:36pm 2

Two methods which I frequently use for debugging: By @smth def memReport(): for obj in gc.get_objects(): if torch.is_tensor(obj): print(type(obj), obj.size()) def cpuStats(): print(sys.version) print(psutil.cpu_percent()) print(psutil.virtual_memory()) # physical memory usage pid = os.getpid() py = psutil.Process(pid) memoryUse = py.memory_info()[0] / 2. ** 30 # memory use in GB...I think print('memory GB:',…

show post in topic

Home
Categories
Guidelines
Terms of Service
Privacy Policy

Powered by Discourse, best viewed with JavaScript enabled