From 2b8d5abb18c829d22f290679b739dc53765198ad Mon Sep 17 00:00:00 2001
From: Amadeusz Szymko <amadeusz.szymko@put.poznan.pl>
Date: Mon, 15 May 2023 09:46:43 +0200
Subject: [PATCH] fix: nvidia arg (#234)

Signed-off-by: Amadeusz Szymko <amadeusz.szymko@put.poznan.pl>
---
 src/rocker/nvidia_extension.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/rocker/nvidia_extension.py b/src/rocker/nvidia_extension.py
index ec0ed5b5..f731f481 100644
--- 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):