-
Notifications
You must be signed in to change notification settings - Fork 63
Adds support for submitting jobs with gpus #794
Conversation
@dPeS - Any chance you can take this for a spin and see if it works in your environment? |
@dposada sure, will do next week and report here when done |
@dposada works as expected... but I see 1 issue here:
is it expected? |
Currently, non-gpu jobs won't get scheduled on agents with gpus (@pschorf correct me if I'm wrong) |
@dPeS That's correct. Currently, we have logic to prevent scheduling jobs without GPUs on GPU-enabled hosts to prevent tying up the non-GPU resources on the host. If that's an issue in your environment we can look into making that configurable. |
@DaoWen any feedback before I merge this? |
@pschorf hi, that would be useful for my case... it's not a bloker since I could spawn p2.xlarge + some non gpu instance, but in future |
Changes proposed in this PR
--gpus
flag tocs submit
--gpus
Why are we making these changes?
This allows users to submit jobs with gpus using
cs
. See issue #793 for more context.