Support for multiple Nvidia GPU selection in terraform

[huyhuyvu01]

Community Note

• Please vote on this issue by adding a 👍 reaction to the original issue to help the community and maintainers prioritize this request
• Please do not leave "+1" or "me too" comments, they generate extra noise for issue followers and do not help prioritize the request
• If you are interested in working on this issue or have submitted a pull request, please leave a comment

Description

Add the ability to support multiple GPU selection in environment when there is multiple GPUs available.
Eg: In my cluster of 8 GPU, i can choose between gpu by using it's index as 2,4,6

New or Affected Resource(s)

• docker_container

Potential Terraform Configuration

resource "docker_container" "base" {
  name    = "foo"
  image   = docker_image.base.latest
  gpus = "1,2,4"
  ....
}

References

[Junkern]

I do not have an nVidia GPU, but did you try it?
Although the documentation says "only all works", the provider should support everything which the docker CLI also supports: https://docs.docker.com/reference/cli/docker/container/run/#gpus

"device=0,2" # should expose the first and third GPU