kaushikbar
commited on
Commit
·
c947eb8
1
Parent(s):
7e719fe
min dall-e
Browse files
app.py
CHANGED
@@ -18,8 +18,7 @@ def filename_from_text(text: str) -> str:
|
|
18 |
return text
|
19 |
|
20 |
def log_gpu_memory():
|
21 |
-
|
22 |
-
pass
|
23 |
|
24 |
log_gpu_memory()
|
25 |
|
|
|
18 |
return text
|
19 |
|
20 |
def log_gpu_memory():
|
21 |
+
print("Date:{}, GPU memory:{}".format(str(datetime.datetime.now()), subprocess.check_output('nvidia-smi').decode('utf-8')))
|
|
|
22 |
|
23 |
log_gpu_memory()
|
24 |
|