-
-
Notifications
You must be signed in to change notification settings - Fork 104
Open
Description
Hi,
I try to create a container with GPU support with fluent docker, as specified in the Docker Documentation.
I figured out, that fluent docker does not have support for the GPU flag. I expected sth. like this:
containerBuilder.UseGpu("all");
I also did not find any way to work around. I expected it being something like this:
containerBuilder.WithCustomArgument("gpus", "all");
// or
containerBuilder.WithCustomArgument("--gpus all");
Am I missing something? The latter one would be some kind of fallback for everybody who needs features you did not provide with the builder.
If you like I would gladly offer to contribute these changes.
Cheers!