fix: make sure gpus are found in local_gpu run

Author: gverkes

src/sagemaker/local/image.py

@@ -860,7 +860,13 @@ def _create_docker_host(
         # to setting --runtime=nvidia in the docker commandline.
         if self.instance_type == "local_gpu":
             host_config["deploy"] = {
-                "resources": {"reservations": {"devices": [{"capabilities": ["gpu"]}]}}
+                "resources": {
+                    "reservations": {
+                        "devices": [
+                            {"count": "all", "capabilities": ["gpu"]}
+                        ]
+                    }
+                }
             }
 
         if not self.is_studio and command == "serve":