谷歌 colab GPU 内存查看与释放

谷歌 colab GPU 内存查看与释放

参考链接

http://thoughtsondl.blogspot.com/2018/06/how-to-release-or-reset-gpu-memory-in.html

# colab
https://colab.research.google.com

1. 查看 GPU 内存占用情况

• 首先安装支持包

  # memory footprint support libraries/code
  !ln -sf /opt/bin/nvidia-smi /usr/bin/nvidia-smi
  !pip install gputil
  !pip install psutil
  !pip install humanize
  

• 然后运行下面的代码

  import psutil
  import humanize
  import os
  import GPUtil as GPU
  
  GPUs = GPU.getGPUs()
  # XXX: only one GPU on Colab and isn’t guaranteed
  gpu = GPUs[0]
  def printm():
      process = psutil.Process(os.getpid())
      print("Gen RAM Free: " + humanize.naturalsize(psutil.virtual_memory().available), " |     Proc size: " + humanize.naturalsize(process.memory_info().rss))
      print("GPU RAM Free: {0:.0f}MB | Used: {1:.0f}MB | Util {2:3.0f}% | Total     {3:.0f}MB".format(gpu.memoryFree, gpu.memoryUsed, gpu.memoryUtil*100, gpu.memoryTotal))
  printm()
  

• 运行结果大致如下所示:

  Gen RAM Free: 12.7 GB  |     Proc size: 117.8 MB
  GPU RAM Free: 11441MB | Used: 0MB | Util   0% | Total     11441MB