Skip to content
master
Switch branches/tags
Go to file
Code

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
Mar 21, 2017

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.

About

Small Python library to automatically set CUDA_VISIBLE_DEVICES to the least loaded device on multi-GPU systems.

Resources

Releases

No releases published

Packages

No packages published

Languages