add more docs

mieskolainen · Oct 11, 2024 · 5721d32 · 5721d32
1 parent aa42827
commit 5721d32
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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.
     """