谷歌 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