Skip to main content

cuthon

Cuthon is a simple Python script to avoid setting CUDA_VISIBLE_DEVICES when running python programs on a GPU. The script will find the first unused GPU(s) then run the program as usual. At its simplest:

cuthon my_program.py

This tool is intended for a fairly niche use case: running python programs on an interactive node which has more than one GPU. For those that have done this often you may be relieved at never having to run nvidia-smi followed by setting CUDA_VISIBLE_DEVICES again (when running a python program that is).

Install

Install with pip:

pip install cuthon

Usage

In general, use cuthon just like you would use python.

  • cuthon to launch a python repl.

  • cuthon -V to see the python version number.

  • cuthon train_model.py to run your program.

For help on available cuthon options type:

cuthon -h --

The output will be:

usage: cuthon.py [-h] [-n NUM_GPUS] [-l]

Select the first unused GPU(s) and run Python. To pass the script arguments
specify '--' between cuthon arguments and arguments to be passed through to
your script. If '--' is not specified, then all arguments will be passed
through.

optional arguments:
  -h, --help            show this help message and exit
  -n NUM_GPUS, --num_gpus NUM_GPUS
                        The number of GPUs to use.
  -l, --least_used      Switch from an unused to a least-used policy.

For example, to run on two available GPUs execute:

cuthon -n 2 -- train_model.py

Metadata

Release files for cuthon 0.5

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for cuthon 0.5
File Size Uploaded Provenance
cuthon-0.5.tar.gz 3.0 kB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for cuthon 0.5
File Interpreter ABI Platform
cuthon-0.5-py2.py3-none-any.whl Python 2, Python 3 none any Details

Total release size: 6.4 kB

Release files / cuthon-0.5.tar.gz

Download URL cuthon-0.5.tar.gz
Size 3.0 kB
Tags Source
SHA-256 checksum
How to use checksums
a6b4cd3eec72db841b864bfaa481bccbf7b186663e0b2f760b6baca8189f1f68
BLAKE2b-256 checksum
How to use checksums
0e08b9e4e1ea51f3b8e9b6b6eedfb0d388b4a9448e709a2648bd13cf0a6a2826
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No

Release files / cuthon-0.5-py2.py3-none-any.whl

Download URL cuthon-0.5-py2.py3-none-any.whl
Size 3.3 kB
Tags Python 2 Python 3
SHA-256 checksum
How to use checksums
7e5e3aa859fe52120f9f2c5ca29dd0fc6b0b3abcd838becae8023b530d8c7c7f
BLAKE2b-256 checksum
How to use checksums
0e49617d50784b9763ab07b2b7dd45fae42a8487ffc1e887d853e4aad5157e27
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No

Security for cuthon 0.5

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page