Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.gitignore
README.md
setGPU.py
setup.py Bump version. Mar 21, 2017

README.md

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 setGPU before any import that will use a GPU like torch or tensorflow.

Dependencies

Licensing

This code is in the public domain.