updage model container start gpu driver

0e94243e · duanjinfei · 3e86446d · 0e94243e
Commit 0e94243e authored Feb 20, 2024 by duanjinfei
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

docker.go operate/docker.go +1 -1

No files found.
--- a/operate/docker.go
+++ b/operate/docker.go
@@ -185,7 +185,7 @@ func (d *DockerOp) CreateContainer(imageName string, dockerCmd *models.DockerCmd
 			DeviceRequests: []container.DeviceRequest{
 				{
 					Driver: "nvidia",
-					Count:  -1, // -1 means all GPUs
+					Count:  1, // -1 means all GPUs
 				},
 			},
 		},