You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Jan 17, 2024. It is now read-only.
Repository navigation
This repository was archived by the owner on Jan 17, 2024. It is now read-only.
[BUG] "Define which GPU(s) will be visible in container" setting fails #38
Thanks for the good plugin to handle GPU resources!
There is a small bug:
"custom - Define which GPU(s) will be visible in container" setting fails to reserve multiple GPUs.
This happens at least when using pipelines.
Hi @mippos I experimented a lot to try and get this to work without success.
I think it might need to fallback from using --gpus to use --runtime=nvidia and setting the environment variable NVIDIA_VISIBLE_DEVICES. This is what happens for older docker versions, but it should work even for the latest docker version. I finally have a dual-GPU machine I can test with, so I'll try to find some time for testing and get this fixed.
Activity
raydouglass commentedon Jul 21, 2021
Hi @mippos I experimented a lot to try and get this to work without success.
I think it might need to fallback from using
--gpusto use--runtime=nvidiaand setting the environment variableNVIDIA_VISIBLE_DEVICES. This is what happens for older docker versions, but it should work even for the latest docker version. I finally have a dual-GPU machine I can test with, so I'll try to find some time for testing and get this fixed.