Benoît Zuber aseedb

aseedb / check_gpu_ram_colab.py

Created February 14, 2020 22:45 — forked from okomarov/check_gpu_ram_colab.py

Check allocated GPU by Colab

	# 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