milyiyo/clean-ram-gpu-colab.py

Created March 31, 2023 17:30

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/milyiyo/bb2542ca5008c1ca206a5cef951deeb2.js"></script>
Save milyiyo/bb2542ca5008c1ca206a5cef951deeb2 to your computer and use it in GitHub Desktop.

Download ZIP

Clean the RAM and GPU in a Colab

Raw

clean-ram-gpu-colab.py

	import gc
	import torch

	gc.collect()

	with torch.no_grad():
	torch.cuda.empty_cache()

	!nvidia-smi

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment