fix: nvidia arg (#234)

Signed-off-by: Amadeusz Szymko <[email protected]>
osrf · May 15, 2023 · 2b8d5ab · 2b8d5ab
1 parent f6d8918
commit 2b8d5ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rocker/nvidia_extension.py b/src/rocker/nvidia_extension.py
@@ -138,7 +138,7 @@ def register_arguments(parser, defaults={}):
             choices=['auto', 'runtime', 'gpus'],
             nargs='?',
             const='auto',
-            default=defaults.get(Nvidia.get_name(), 'auto'),
+            default=defaults.get(Nvidia.get_name(), None),
             help="Enable nvidia. Default behavior is to pick flag based on docker version.")
 
 class Cuda(RockerExtension):