POD-819: Add --gpus=all if host requirments contain GPU (#1229)

* POD-819: Add --gpus=all if host requirments contain GPU

* POD-819: Add nill check

pkg/driver/docker/docker.go

@@ -6,6 +6,7 @@ import (
 	"io"
 	"os"
 	"runtime"
+	"slices"
 	"strconv"
 	"strings"
 
@@ -325,6 +326,17 @@ func (d *dockerDriver) RunDockerDevContainer(
 	}
 
 	// runArgs
+	// check if we need to add --gpus=all to the run args based on the dev container's host requirments
+	if parsedConfig.HostRequirements != nil {
+		usesGpu, err := parsedConfig.HostRequirements.GPU.Bool()
+		if err != nil && usesGpu {
+			// check if the user manually add --gpus=all, if not then add it
+			if !slices.Contains(parsedConfig.RunArgs, "--gpus=all") {
+				args = append(args, "--gpus=all")
+			}
+		}
+	}
+
 	args = append(args, parsedConfig.RunArgs...)
 
 	// run detached