Skip to content

bamos/setGPU

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 

setGPU

A small Python library that automatically sets CUDA_VISIBLE_DEVICES to the least-loaded GPU on multi-GPU systems and can be used by:

  1. Putting import setGPU before any import that will use a GPU like Torch, TensorFlow, or JAX.
  2. Defining an alias such as alias setGPU='eval $(python3 -m setGPU)' and calling that to set the GPU in the shell before running another program that uses the GPU.

Installation

pip install git+https://github.com/bamos/setGPU.git

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

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages