A small Python library that automatically sets CUDA_VISIBLE_DEVICES
to the least-loaded GPU on multi-GPU systems and can be used by:
from setgpu import set_gpu
set_gpu()pip install git+https://github.com/datthinh1801/setGPU.git
- Jongwook Choi's gpustat library (
pip install gpustat)
This code is in the public domain.