I'm working with TensorFlow framework on multi GPU machine. I want to know that, is there any utility available to monitor TensorFlow memory usage for CPU, RAM and GPU memory consumption. I'm using following software and hardware:

  • 6 AMD GPUs (RX470-Sapphire)
  • Ubuntu 16.04.5 LTS
  • Linux 4.15.0-45-generic x86_64
  • ROCm driver
  • TensorFlow-rocm
  • Python3.5

I know about "rocm-smi" that shows only GPU consumption. But i want to see Native memory call. Kindly let me know if someone know this kind of utility.

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Browse other questions tagged or ask your own question.