nikhilweee
/ dummy_gpu.py
Created
April 24, 2022 20:28
Run dummy GPU job whenever usage drops below 5%
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import torch | |
| import subprocess | |
| import time | |
| import logging | |
| # Takes about 8GB | |
| ndim = 25_000 | |
| logging.basicConfig(format='[%(asctime)s] %(filename)s [%(levelname).1s] %(message)s', level=logging.DEBUG) | |
| def get_gpu_usage(): |
OlderNewer