tensorchord/envd

feat: the "evnd up" command adds a gpu-set parameter that looks like "docker run --gpus"

arugal opened this issue · 2 comments

Describe the feature

The evnd up command adds a gpu-set parameter that looks like docker run-gpus and supports the following forms of parameters

  1. all
  2. device=0,2
  3. 0,1

Why do you need this feature?

The container can be assigned specific GPU resources.

Additional context

No response

I have already implemented it on my own branch and have been using it for some time. I would like to contribute it to envd :)

I have already implemented it on my own branch and have been using it for some time. I would like to contribute it to envd :)

You can create a PR.