A simple tool to select the optimal CUDA device based on memory, power, or utilization.
Project description
Auto Cuda Selector
A simple tool to select the optimal CUDA device based on memory, power, or utilization.
Install
pip install cuda-selector
Usage
from cuda_selector import auto_cuda
# Select cuda device with most memory available
device = auto_cuda()
# Select cuda device with lowest power usasge usage
device = auto_cuda('power')
# Select cuda device with lowest power usasge usage
device = auto_cuda('utilization')
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
cuda_selector-0.1.1.tar.gz
(3.0 kB
view hashes)