bug: local_gpu gives error Unknown runtime specified nvidia

Author: teticio

src/sagemaker/local/image.py

@@ -746,7 +746,6 @@ def _create_docker_host(self, host, environment, optml_subdirs, command, volumes
             "volumes": [v.map for v in optml_volumes],
             "environment": environment,
             "networks": {"sagemaker-local": {"aliases": [host]}},
-            "deploy": {"resources": {"reservations": {"devices": [{"capabilities": ["gpu"]}]}}},
         }
 
         if command != "process":
@@ -760,7 +759,7 @@ def _create_docker_host(self, host, environment, optml_subdirs, command, volumes
         # for GPU support pass in nvidia as the runtime, this is equivalent
         # to setting --runtime=nvidia in the docker commandline.
         if self.instance_type == "local_gpu":
-            host_config["runtime"] = "nvidia"
+            host_config["deploy"] = {"resources": {"reservations": {"devices": [{"capabilities": ["gpu"]}]}}}
 
         if command == "serve":
             serving_port = (