From 2154a34f3f5f5322ddb3fa23ce934cb483bd094d Mon Sep 17 00:00:00 2001
From: Jacob Howard <jacob.howard@docker.com>
Date: Tue, 13 May 2025 16:54:05 -0600
Subject: [PATCH] standalone: use nvidia runtime if using GPU

Signed-off-by: Jacob Howard <jacob.howard@docker.com>
---
 pkg/standalone/containers.go | 11 ++---------
 1 file changed, 2 insertions(+), 9 deletions(-)

diff --git a/pkg/standalone/containers.go b/pkg/standalone/containers.go
index 29c7f9a1..1287b3eb 100644
--- a/pkg/standalone/containers.go
+++ b/pkg/standalone/containers.go
@@ -73,15 +73,8 @@ func CreateControllerContainer(ctx context.Context, dockerClient *client.Client,
 		},
 	}
 	if gpu {
-		hostConfig.Resources = container.Resources{
-			DeviceRequests: []container.DeviceRequest{
-				{
-					Driver:       "nvidia",
-					Count:        -1,
-					Capabilities: [][]string{{"gpu"}},
-				},
-			},
-		}
+		hostConfig.Runtime = "nvidia"
+		hostConfig.DeviceRequests = []container.DeviceRequest{{Count: -1, Capabilities: [][]string{{"gpu"}}}}
 	}
 
 	// Create the container.