No public description

PiperOrigin-RevId: 619570154
googlecolab · Mar 27, 2024 · 1d5784e · 1d5784e
1 parent dd7d19d
commit 1d5784e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/google/colab/_serverextension/_resource_monitor.py b/google/colab/_serverextension/_resource_monitor.py
@@ -18,6 +18,8 @@
 # Track whether user has used the GPU in the current session.
 _GPU_EVER_USED = False
 
+_T4_TOTAL_MEMORY_BYTES = 16106127360
+
 
 @functools.cache
 def _nvidia_smi_path():
@@ -48,10 +50,10 @@ def get_gpu_stats():
     return [
         GpuInfo(
             name='Tesla T4',
-            memoryUsedBytes=123,
-            memoryTotalBytes=456,
+            memoryUsedBytes=0.35 * _T4_TOTAL_MEMORY_BYTES,
+            memoryTotalBytes=_T4_TOTAL_MEMORY_BYTES,
             gpuUtilization=0.1,
-            memoryUtilization=0.2,
+            memoryUtilization=0.35,
             everUsed=True,
         )
     ]