updage model container start gpu driver

operate/docker.go

@@ -176,6 +176,8 @@ func (d *DockerOp) CreateContainer(imageName string, dockerCmd *models.DockerCmd
 	portMap = make(map[nat.Port][]nat.PortBinding, 0)
 	portMap[nat.Port(dockerCmd.ContainerPort+"/tcp")] = portBinds
 
+	capabilities := make([][]string, 0)
+	capabilities = append(capabilities, []string{"gpu"})
 	resp, err := d.dockerClient.ContainerCreate(ctx, &container.Config{
 		Image: imageName,
 	}, &container.HostConfig{
@@ -184,8 +186,9 @@ func (d *DockerOp) CreateContainer(imageName string, dockerCmd *models.DockerCmd
 		Resources: container.Resources{
 			DeviceRequests: []container.DeviceRequest{
 				{
-					Driver: "nvidia",
-					Count:  1, // -1 means all GPUs
+					//Driver:       "nvidia",
+					Count:        -1, // -1 means all GPUs
+					Capabilities: capabilities,
 				},
 			},
 		},