Update bytes to GB with bitshift (#6886)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-03-07 19:26:37 +01:00
1 parent acc58c1dcf
commit e6e36aac10
4 files changed
+11 -10

No files matched your search

+3 -4
View File
@@ -21,14 +21,13 @@ def notebook_init(verbose=True):
if is_colab():
shutil.rmtree('/content/sample_data', ignore_errors=True) # remove colab /sample_data directory
# System info
if verbose:
# System info
# gb = 1 / 1000 ** 3 # bytes to GB
gib = 1 / 1024 ** 3 # bytes to GiB
gb = 1 << 30 # bytes to GiB (1024 ** 3)
ram = psutil.virtual_memory().total
total, used, free = shutil.disk_usage("/")
display.clear_output()
s = f'({os.cpu_count()} CPUs, {ram * gib:.1f} GB RAM, {(total - free) * gib:.1f}/{total * gib:.1f} GB disk)'
s = f'({os.cpu_count()} CPUs, {ram / gb:.1f} GB RAM, {(total - free) / gb:.1f}/{total / gb:.1f} GB disk)'
else:
s = ''