setGPU
A small Python library that automatically sets CUDA_VISIBLE_DEVICES
to the least-loaded GPU on multi-GPU systems.
- Installation:
pip install setGPU - Usage:
import setGPUbefore any import that will use a GPU liketorchortensorflow.
Dependencies
- Jongwook Choi's gpustat library.
Licensing
This code is in the public domain.