diff --git a/docs/source/notes/installation.rst b/docs/source/notes/installation.rst
index 01034800..460d6b2a 100644
--- a/docs/source/notes/installation.rst
+++ b/docs/source/notes/installation.rst
@@ -23,7 +23,7 @@ Then execute the installer with ``bash filename.sh`` and finally set ``.condarc`
       - defaults
 
 
-Pre-installed CUDA paths (EXPERIMENTAL)
+Pre-installed CUDA paths (OBSOLETE)
 ------------------------------------
 .. code-block:: none
 
diff --git a/icenet/tools/io.py b/icenet/tools/io.py
index d3e38681..f886b831 100644
--- a/icenet/tools/io.py
+++ b/icenet/tools/io.py
@@ -219,7 +219,7 @@ def get_gpu_memory_map():
         print(f'Error: Could not run nvidia-smi: {e}')
         return None
 
-def torch_cuda_total_memory(device):
+def torch_cuda_total_memory(device='cuda:0'):
     """
     Return CUDA device VRAM available in GB.
     """