Further testing and looking at the source code confirms what looks to me like a bug in Slurm. GPUs that are not configured in gres.conf are detected by slurmd in the system and discarded since they aren't found in gres.conf. That's fine except they should also be hidden through cgroup control so that they aren't visible along with allocated GPUs when a job is run. Slurm assumes that the job can only see the GPUs that it allocates to the job and sets the $CUDA_VISIBLE_DEVICES accordingly. Unfortunately, the job actually sees the allocated GPUs plus any unconfigured GPUs and $CUDA_VISIBLE_DEVICES may or may not happen to correspond to the GPU(s) allocated by Slurm.
I was hoping that I could write a Prolog script that would adjust $CUDA_VISIBLE_DEVICES to remove any unconfigured GPUs but any changes using "export CUDA_VISIBLE_DEVICES=..." don't seem to have an effect upon the actual environment of the job. Steve ________________________________ From: Wilson, Steven M <ste...@purdue.edu> Sent: Friday, July 14, 2023 4:10 PM To: slurm-users@lists.schedmd.com <slurm-users@lists.schedmd.com> Subject: Re: [slurm-users] Unconfigured GPUs being allocated It's not so much whether a job may or may not access the GPU but rather which GPU(s) is(are) included in $CUDA_VISIBLE_DEVICES. That is what controls what our CUDA jobs can see and therefore use (within any cgroups constraints, of course). In my case, Slurm is sometimes setting $CUDA_VISIBLE_DEVICES to a GPU that is not in the Slurm configuration because it is intended only for driving the display and not GPU computations. Thanks for your thoughts! Steve ________________________________ From: slurm-users <slurm-users-boun...@lists.schedmd.com> on behalf of Christopher Samuel <ch...@csamuel.org> Sent: Friday, July 14, 2023 1:57 PM To: slurm-users@lists.schedmd.com <slurm-users@lists.schedmd.com> Subject: Re: [slurm-users] Unconfigured GPUs being allocated [You don't often get email from ch...@csamuel.org. Learn why this is important at https://aka.ms/LearnAboutSenderIdentification ] ---- External Email: Use caution with attachments, links, or sharing data ---- On 7/14/23 10:20 am, Wilson, Steven M wrote: > I upgraded Slurm to 23.02.3 but I'm still running into the same problem. > Unconfigured GPUs (those absent from gres.conf and slurm.conf) are still > being made available to jobs so we end up with compute jobs being run on > GPUs which should only be used I think this is expected - it's not that Slurm is making them available, it's that it's unaware of them and so doesn't control them in the way it does for the GPUs it does know about. So you get the default behaviour (any process can access them). If you want to stop them being accessed from Slurm you'd need to find a way to prevent that access via cgroups games or similar. All the best, Chris -- Chris Samuel : https://nam04.safelinks.protection.outlook.com/?url=http%3A%2F%2Fwww.csamuel.org%2F&data=05%7C01%7Cstevew%40purdue.edu%7C6fba97485b73413521d208db8494160a%7C4130bd397c53419cb1e58758d6d63f21%7C0%7C0%7C638249543794377751%7CUnknown%7CTWFpbGZsb3d8eyJWIjoiMC4wLjAwMDAiLCJQIjoiV2luMzIiLCJBTiI6Ik1haWwiLCJXVCI6Mn0%3D%7C3000%7C%7C%7C&sdata=VslW51ree1Ibt3xfYyy99Aj%2BREZh7BqpM6Ipg3jAM84%3D&reserved=0<http://www.csamuel.org/> : Berkeley, CA, USA
