keep-gpu 0.1.0.post2__tar.gz

keep_gpu-0.1.0.post2/AUTHORS.rst

@@ -0,0 +1,13 @@
+=======
+Credits
+=======
+
+Development Lead
+----------------
+
+* Siyuan Wang <sywang0227@gmail.com>
+
+Contributors
+------------
+
+None yet. Why not be the first?

keep_gpu-0.1.0.post2/CONTRIBUTING.rst

@@ -0,0 +1,137 @@
+.. highlight:: shell
+
+============
+Contributing
+============
+
+Contributions are welcome, and they are greatly appreciated! Every little bit
+helps, and credit will always be given.
+
+You can contribute in many ways:
+
+Types of Contributions
+----------------------
+
+Report Bugs
+~~~~~~~~~~~
+
+Report bugs at https://github.com/Wangmerlyn/keep_gpu/issues.
+
+If you are reporting a bug, please include:
+
+* Your operating system name and version.
+* Any details about your local setup that might be helpful in troubleshooting.
+* Detailed steps to reproduce the bug.
+
+Fix Bugs
+~~~~~~~~
+
+Look through the GitHub issues for bugs. Anything tagged with "bug" and "help
+wanted" is open to whoever wants to implement it.
+
+Implement Features
+~~~~~~~~~~~~~~~~~~
+
+Look through the GitHub issues for features. Anything tagged with "enhancement"
+and "help wanted" is open to whoever wants to implement it.
+
+Write Documentation
+~~~~~~~~~~~~~~~~~~~
+
+Keep GPU could always use more documentation, whether as part of the
+official Keep GPU docs, in docstrings, or even on the web in blog posts,
+articles, and such.
+
+Submit Feedback
+~~~~~~~~~~~~~~~
+
+The best way to send feedback is to file an issue at https://github.com/Wangmerlyn/keep_gpu/issues.
+
+If you are proposing a feature:
+
+* Explain in detail how it would work.
+* Keep the scope as narrow as possible, to make it easier to implement.
+* Remember that this is a volunteer-driven project, and that contributions
+  are welcome :)
+
+Get Started!
+------------
+
+Ready to contribute? Here's how to set up `keep_gpu` for local development.
+
+1. Fork the `keep_gpu` repo on GitHub.
+2. Clone your fork locally::
+
+    $ git clone git@github.com:your_name_here/keep_gpu.git
+
+3. Install your local copy into a virtualenv. Assuming you have virtualenvwrapper installed, this is how you set up your fork for local development::
+
+    $ mkvirtualenv keep_gpu
+    $ cd keep_gpu/
+    $ python setup.py develop
+
+4. Create a branch for local development::
+
+    $ git checkout -b name-of-your-bugfix-or-feature
+
+   Now you can make your changes locally.
+
+5. When you're done making changes, check that your changes pass flake8 and the
+   tests, including testing other Python versions with tox::
+
+    $ make lint
+    $ make test
+    Or
+    $ make test-all
+
+   To get flake8 and tox, just pip install them into your virtualenv.
+
+6. Commit your changes and push your branch to GitHub::
+
+    $ git add .
+    $ git commit -m "Your detailed description of your changes."
+    $ git push origin name-of-your-bugfix-or-feature
+
+7. Submit a pull request through the GitHub website.
+
+Pull Request Guidelines
+-----------------------
+
+Before you submit a pull request, check that it meets these guidelines:
+
+1. The pull request should include tests.
+2. If the pull request adds functionality, the docs should be updated. Put
+   your new functionality into a function with a docstring, and add the
+   feature to the list in README.rst.
+3. The pull request should work for Python 3.5, 3.6, 3.7 and 3.8, and for PyPy. Check
+   https://travis-ci.com/Wangmerlyn/keep_gpu/pull_requests
+   and make sure that the tests pass for all supported Python versions.
+
+Tips
+----
+
+To run a subset of tests::
+
+
+    $ python -m unittest tests.test_keep_gpu
+
+Deploying
+---------
+
+A reminder for the maintainers on how to deploy.
+Make sure all your changes are committed (including an entry in HISTORY.rst).
+Then run::
+
+$ bump2version patch # possible: major / minor / patch
+$ git push
+$ git push --tags
+
+Travis will then deploy to PyPI if tests pass.
+
+Code of Conduct
+---------------
+
+Please note that this project is released with a `Contributor Code of Conduct`_.
+By participating in this project you agree to abide by its terms.
+
+.. _`Contributor Code of Conduct`: CODE_OF_CONDUCT.rst

keep_gpu-0.1.0.post2/HISTORY.rst

@@ -0,0 +1,8 @@
+=======
+History
+=======
+
+0.1.0 (2025-05-20)
+------------------
+
+* First release on PyPI.

keep_gpu-0.1.0.post2/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Wang Siyuan
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

keep_gpu-0.1.0.post2/MANIFEST.in

@@ -0,0 +1,11 @@
+include AUTHORS.rst
+include CONTRIBUTING.rst
+include HISTORY.rst
+include LICENSE
+include README.rst
+
+recursive-include tests *
+recursive-exclude * __pycache__
+recursive-exclude * *.py[co]
+
+recursive-include docs *.rst conf.py Makefile make.bat *.jpg *.png *.gif

keep_gpu-0.1.0.post2/PKG-INFO

@@ -0,0 +1,62 @@
+Metadata-Version: 2.4
+Name: keep_gpu
+Version: 0.1.0.post2
+Summary: Keep GPU is a simple cli app that keep your gpus running
+Author-email: Siyuan Wang <sywang0227@gmail.com>
+Maintainer-email: Siyuan Wang <sywang0227@gmail.com>
+License: MIT license
+Project-URL: bugs, https://github.com/Wangmerlyn/keep_gpu/issues
+Project-URL: changelog, https://github.com/Wangmerlyn/keep_gpu/blob/master/changelog.md
+Project-URL: homepage, https://github.com/Wangmerlyn/keep_gpu
+Description-Content-Type: text/x-rst
+License-File: LICENSE
+License-File: AUTHORS.rst
+Requires-Dist: typer
+Provides-Extra: dev
+Requires-Dist: coverage; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+Dynamic: license-file
+
+========
+Keep GPU
+========
+
+
+.. image:: https://img.shields.io/pypi/v/keep_gpu.svg
+        :target: https://pypi.python.org/pypi/keep_gpu
+
+.. image:: https://img.shields.io/travis/Wangmerlyn/KeepGPU.svg
+        :target: https://travis-ci.com/Wangmerlyn/KeepGPU
+
+.. image:: https://readthedocs.org/projects/keep-gpu/badge/?version=latest
+        :target: https://keep-gpu.readthedocs.io/en/latest/?version=latest
+        :alt: Documentation Status
+
+
+.. image:: https://pyup.io/repos/github/Wangmerlyn/keep_gpu/shield.svg
+     :target: https://pyup.io/repos/github/Wangmerlyn/keep_gpu/
+     :alt: Updates
+
+
+
+Keep GPU is a simple cli app that keep your gpus running
+
+
+* Free software: MIT license
+* Documentation: https://keep-gpu.readthedocs.io.
+
+
+Features
+--------
+
+* TODO
+
+Credits
+-------
+
+This package was created with Cookiecutter_ and the `audreyr/cookiecutter-pypackage`_ project template.
+
+.. _Cookiecutter: https://github.com/audreyr/cookiecutter
+.. _`audreyr/cookiecutter-pypackage`: https://github.com/audreyr/cookiecutter-pypackage

keep_gpu-0.1.0.post2/README.rst

@@ -0,0 +1,41 @@
+========
+Keep GPU
+========
+
+
+.. image:: https://img.shields.io/pypi/v/keep_gpu.svg
+        :target: https://pypi.python.org/pypi/keep_gpu
+
+.. image:: https://img.shields.io/travis/Wangmerlyn/KeepGPU.svg
+        :target: https://travis-ci.com/Wangmerlyn/KeepGPU
+
+.. image:: https://readthedocs.org/projects/keep-gpu/badge/?version=latest
+        :target: https://keep-gpu.readthedocs.io/en/latest/?version=latest
+        :alt: Documentation Status
+
+
+.. image:: https://pyup.io/repos/github/Wangmerlyn/keep_gpu/shield.svg
+     :target: https://pyup.io/repos/github/Wangmerlyn/keep_gpu/
+     :alt: Updates
+
+
+
+Keep GPU is a simple cli app that keep your gpus running
+
+
+* Free software: MIT license
+* Documentation: https://keep-gpu.readthedocs.io.
+
+
+Features
+--------
+
+* TODO
+
+Credits
+-------
+
+This package was created with Cookiecutter_ and the `audreyr/cookiecutter-pypackage`_ project template.
+
+.. _Cookiecutter: https://github.com/audreyr/cookiecutter
+.. _`audreyr/cookiecutter-pypackage`: https://github.com/audreyr/cookiecutter-pypackage

keep_gpu-0.1.0.post2/docs/Makefile

@@ -0,0 +1,20 @@
+# Minimal makefile for Sphinx documentation
+#
+
+# You can set these variables from the command line.
+SPHINXOPTS    =
+SPHINXBUILD   = python -msphinx
+SPHINXPROJ    = keep_gpu
+SOURCEDIR     = .
+BUILDDIR      = _build
+
+# Put it first so that "make" without argument is like "make help".
+help:
+	@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
+
+.PHONY: help Makefile
+
+# Catch-all target: route all unknown targets to Sphinx using the new
+# "make mode" option.  $(O) is meant as a shortcut for $(SPHINXOPTS).
+%: Makefile
+	@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)

keep_gpu-0.1.0.post2/docs/authors.rst

@@ -0,0 +1 @@
+.. include:: ../AUTHORS.rst

keep_gpu-0.1.0.post2/docs/conf.py

@@ -0,0 +1,162 @@
+#!/usr/bin/env python
+#
+# keep_gpu documentation build configuration file, created by
+# sphinx-quickstart on Fri Jun  9 13:47:02 2017.
+#
+# This file is execfile()d with the current directory set to its
+# containing dir.
+#
+# Note that not all possible configuration values are present in this
+# autogenerated file.
+#
+# All configuration values have a default; values that are commented out
+# serve to show the default.
+
+# If extensions (or modules to document with autodoc) are in another
+# directory, add these directories to sys.path here. If the directory is
+# relative to the documentation root, use os.path.abspath to make it
+# absolute, like shown here.
+#
+import os
+import sys
+sys.path.insert(0, os.path.abspath('..'))
+
+import keep_gpu
+
+# -- General configuration ---------------------------------------------
+
+# If your documentation needs a minimal Sphinx version, state it here.
+#
+# needs_sphinx = '1.0'
+
+# Add any Sphinx extension module names here, as strings. They can be
+# extensions coming with Sphinx (named 'sphinx.ext.*') or your custom ones.
+extensions = ['sphinx.ext.autodoc', 'sphinx.ext.viewcode']
+
+# Add any paths that contain templates here, relative to this directory.
+templates_path = ['_templates']
+
+# The suffix(es) of source filenames.
+# You can specify multiple suffix as a list of string:
+#
+# source_suffix = ['.rst', '.md']
+source_suffix = '.rst'
+
+# The master toctree document.
+master_doc = 'index'
+
+# General information about the project.
+project = 'Keep GPU'
+copyright = "2025, Siyuan Wang"
+author = "Siyuan Wang"
+
+# The version info for the project you're documenting, acts as replacement
+# for |version| and |release|, also used in various other places throughout
+# the built documents.
+#
+# The short X.Y version.
+version = keep_gpu.__version__
+# The full version, including alpha/beta/rc tags.
+release = keep_gpu.__version__
+
+# The language for content autogenerated by Sphinx. Refer to documentation
+# for a list of supported languages.
+#
+# This is also used if you do content translation via gettext catalogs.
+# Usually you set "language" from the command line for these cases.
+language = None
+
+# List of patterns, relative to source directory, that match files and
+# directories to ignore when looking for source files.
+# This patterns also effect to html_static_path and html_extra_path
+exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store']
+
+# The name of the Pygments (syntax highlighting) style to use.
+pygments_style = 'sphinx'
+
+# If true, `todo` and `todoList` produce output, else they produce nothing.
+todo_include_todos = False
+
+
+# -- Options for HTML output -------------------------------------------
+
+# The theme to use for HTML and HTML Help pages.  See the documentation for
+# a list of builtin themes.
+#
+html_theme = 'alabaster'
+
+# Theme options are theme-specific and customize the look and feel of a
+# theme further.  For a list of options available for each theme, see the
+# documentation.
+#
+# html_theme_options = {}
+
+# Add any paths that contain custom static files (such as style sheets) here,
+# relative to this directory. They are copied after the builtin static files,
+# so a file named "default.css" will overwrite the builtin "default.css".
+html_static_path = ['_static']
+
+
+# -- Options for HTMLHelp output ---------------------------------------
+
+# Output file base name for HTML help builder.
+htmlhelp_basename = 'keep_gpudoc'
+
+
+# -- Options for LaTeX output ------------------------------------------
+
+latex_elements = {
+    # The paper size ('letterpaper' or 'a4paper').
+    #
+    # 'papersize': 'letterpaper',
+
+    # The font size ('10pt', '11pt' or '12pt').
+    #
+    # 'pointsize': '10pt',
+
+    # Additional stuff for the LaTeX preamble.
+    #
+    # 'preamble': '',
+
+    # Latex figure (float) alignment
+    #
+    # 'figure_align': 'htbp',
+}
+
+# Grouping the document tree into LaTeX files. List of tuples
+# (source start file, target name, title, author, documentclass
+# [howto, manual, or own class]).
+latex_documents = [
+    (master_doc, 'keep_gpu.tex',
+     'Keep GPU Documentation',
+     'Siyuan Wang', 'manual'),
+]
+
+
+# -- Options for manual page output ------------------------------------
+
+# One entry per manual page. List of tuples
+# (source start file, name, description, authors, manual section).
+man_pages = [
+    (master_doc, 'keep_gpu',
+     'Keep GPU Documentation',
+     [author], 1)
+]
+
+
+# -- Options for Texinfo output ----------------------------------------
+
+# Grouping the document tree into Texinfo files. List of tuples
+# (source start file, target name, title, author,
+#  dir menu entry, description, category)
+texinfo_documents = [
+    (master_doc, 'keep_gpu',
+     'Keep GPU Documentation',
+     author,
+     'keep_gpu',
+     'One line description of project.',
+     'Miscellaneous'),
+]
+
+
+

keep_gpu-0.1.0.post2/docs/contributing.rst

@@ -0,0 +1 @@
+.. include:: ../CONTRIBUTING.rst

keep_gpu-0.1.0.post2/docs/history.rst

@@ -0,0 +1 @@
+.. include:: ../HISTORY.rst

keep_gpu-0.1.0.post2/docs/index.rst

@@ -0,0 +1,20 @@
+Welcome to Keep GPU's documentation!
+======================================
+
+.. toctree::
+   :maxdepth: 2
+   :caption: Contents:
+
+   readme
+   installation
+   usage
+   modules
+   contributing
+   authors
+   history
+
+Indices and tables
+==================
+* :ref:`genindex`
+* :ref:`modindex`
+* :ref:`search`

keep_gpu-0.1.0.post2/docs/installation.rst

@@ -0,0 +1,51 @@
+.. highlight:: shell
+
+============
+Installation
+============
+
+
+Stable release
+--------------
+
+To install Keep GPU, run this command in your terminal:
+
+.. code-block:: console
+
+    $ pip install keep_gpu
+
+This is the preferred method to install Keep GPU, as it will always install the most recent stable release.
+
+If you don't have `pip`_ installed, this `Python installation guide`_ can guide
+you through the process.
+
+.. _pip: https://pip.pypa.io
+.. _Python installation guide: http://docs.python-guide.org/en/latest/starting/installation/
+
+
+From sources
+------------
+
+The sources for Keep GPU can be downloaded from the `Github repo`_.
+
+You can either clone the public repository:
+
+.. code-block:: console
+
+    $ git clone git://github.com/Wangmerlyn/keep_gpu
+
+Or download the `tarball`_:
+
+.. code-block:: console
+
+    $ curl -OJL https://github.com/Wangmerlyn/keep_gpu/tarball/master
+
+Once you have a copy of the source, you can install it with:
+
+.. code-block:: console
+
+    $ python setup.py install
+
+
+.. _Github repo: https://github.com/Wangmerlyn/keep_gpu
+.. _tarball: https://github.com/Wangmerlyn/keep_gpu/tarball/master

keep_gpu-0.1.0.post2/docs/make.bat

@@ -0,0 +1,36 @@
+@ECHO OFF
+
+pushd %~dp0
+
+REM Command file for Sphinx documentation
+
+if "%SPHINXBUILD%" == "" (
+	set SPHINXBUILD=python -msphinx
+)
+set SOURCEDIR=.
+set BUILDDIR=_build
+set SPHINXPROJ=keep_gpu
+
+if "%1" == "" goto help
+
+%SPHINXBUILD% >NUL 2>NUL
+if errorlevel 9009 (
+	echo.
+	echo.The Sphinx module was not found. Make sure you have Sphinx installed,
+	echo.then set the SPHINXBUILD environment variable to point to the full
+	echo.path of the 'sphinx-build' executable. Alternatively you may add the
+	echo.Sphinx directory to PATH.
+	echo.
+	echo.If you don't have Sphinx installed, grab it from
+	echo.http://sphinx-doc.org/
+	exit /b 1
+)
+
+%SPHINXBUILD% -M %1 %SOURCEDIR% %BUILDDIR% %SPHINXOPTS%
+goto end
+
+:help
+%SPHINXBUILD% -M help %SOURCEDIR% %BUILDDIR% %SPHINXOPTS%
+
+:end
+popd

keep_gpu-0.1.0.post2/docs/readme.rst

@@ -0,0 +1 @@
+.. include:: ../README.rst

keep_gpu-0.1.0.post2/docs/usage.rst

@@ -0,0 +1,7 @@
+=====
+Usage
+=====
+
+To use Keep GPU in a project::
+
+    import keep_gpu

keep_gpu-0.1.0.post2/pyproject.toml

@@ -0,0 +1,66 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "keep_gpu"
+version = "0.1.0.post2"
+description = "Keep GPU is a simple cli app that keep your gpus running"
+readme = "README.rst"
+authors = [
+  {name = "Siyuan Wang", email = "sywang0227@gmail.com"}
+]
+maintainers = [
+  {name = "Siyuan Wang", email = "sywang0227@gmail.com"}
+]
+classifiers = [
+
+]
+license = {text = "MIT license"}
+dependencies = [
+  "typer"
+]
+
+[project.scripts]
+keep-gpu = "keep_gpu.cli:main"
+
+[project.optional-dependencies]
+dev = [
+    "coverage",  # testing
+    "mypy",  # linting
+    "pytest",  # testing
+    "ruff"  # linting
+]
+
+[project.urls]
+
+bugs = "https://github.com/Wangmerlyn/keep_gpu/issues"
+changelog = "https://github.com/Wangmerlyn/keep_gpu/blob/master/changelog.md"
+homepage = "https://github.com/Wangmerlyn/keep_gpu"
+
+[tool.setuptools]
+package-dir = {"" = "src"}
+
+[tool.setuptools.package-data]
+"*" = ["*.*"]
+
+
+
+
+# Mypy
+# ----
+
+[tool.mypy]
+files = "."
+
+# Use strict defaults
+strict = true
+warn_unreachable = true
+warn_no_return = true
+
+[[tool.mypy.overrides]]
+# Don't require test functions to include types
+module = "tests.*"
+allow_untyped_defs = true
+disable_error_code = "attr-defined"
+

keep_gpu-0.1.0.post2/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

keep_gpu-0.1.0.post2/src/keep_gpu/__init__.py

@@ -0,0 +1,5 @@
+"""Top-level package for Keep GPU."""
+
+__author__ = """Siyuan Wang"""
+__email__ = 'sywang0227@gmail.com'
+__version__ = '0.1.0.post2'

keep_gpu-0.1.0.post2/src/keep_gpu/benchmark.py

@@ -0,0 +1,51 @@
+import torch
+import time
+import subprocess
+import re
+import random
+
+def get_gpu_util(rank):
+    cmds = ['nvidia-smi', '-i', str(rank)]
+    proc = subprocess.Popen(cmds, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    stdout, stderr = proc.communicate()
+    outputs = stdout.decode('utf-8').split('\n')
+
+    util = 0
+    for output in outputs[::-1]:
+        if 'Default' in output:
+            util = int(re.findall(r'\d+', output)[-1])
+            break
+    else:
+        print(f"rank {rank}: couldn't match any, check GPU status!")
+    return util
+
+def keep(rank, args):
+    torch.cuda.set_device(rank)
+    print(f'benchmarking {args.gpus} gpus...')
+    while True:
+        n = random.randint(5, 9)
+        a = torch.rand((8192 * n, 8192)).cuda()
+        b = torch.rand((8192 * n, 8192)).cuda()
+
+        tic = time.time()
+        for _ in range(5000):
+            c = a * b
+        torch.cuda.synchronize()
+        toc = time.time()
+        if rank == 0:
+            print('benchmark 5K matmul: time span: {}ms'.format((toc - tic) * 1000 / 5000))
+
+        time.sleep(args.interval)
+        while get_gpu_util(rank) > 10:
+            print(f'rank {rank}: GPU busy, sleeping...')
+            time.sleep(args.interval)
+        print(f'rank {rank} resumes')
+
+def run_benchmark(gpus=1, interval=100):
+    class Args:
+        def __init__(self, gpus, interval):
+            self.gpus = gpus
+            self.interval = interval
+
+    args = Args(gpus, interval)
+    torch.multiprocessing.spawn(keep, args=(args,), nprocs=gpus, join=True)

keep_gpu-0.1.0.post2/src/keep_gpu/cli.py

@@ -0,0 +1,23 @@
+"""Console script for keep_gpu."""
+import keep_gpu
+from .keep_gpu import run
+
+import typer
+from rich.console import Console
+
+app = typer.Typer()
+console = Console()
+
+
+@app.command()
+def main():
+    """Console script for keep_gpu."""
+    console.print("Replace this message by putting your code into "
+               "keep_gpu.cli.main")
+    console.print("See Typer documentation at https://typer.tiangolo.com/")
+    run()
+    
+
+
+if __name__ == "__main__":
+    app()

keep_gpu-0.1.0.post2/src/keep_gpu/keep_gpu.py

@@ -0,0 +1,27 @@
+import subprocess
+import time
+import os
+import torch
+
+from .benchmark import run_benchmark
+
+def check_gpu_usage():
+    result = subprocess.run(['nvidia-smi', '--query-gpu=utilization.gpu', '--format=csv,noheader'], capture_output=True, text=True)
+    gpu_usages = [int(line.split()[0]) for line in result.stdout.strip().split('\n')]
+    return any(usage > 0 for usage in gpu_usages)
+
+def run():
+    idle_count = 0
+    gpu_count = torch.cuda.device_count()
+    print(f'GPU count: {gpu_count}')
+    while True:
+        if not check_gpu_usage():
+            idle_count += 1
+        else:
+            idle_count = 0
+
+        if idle_count >= 1:
+            run_benchmark(gpu_count)
+            idle_count = 0
+
+        time.sleep(300)

keep_gpu-0.1.0.post2/src/keep_gpu.egg-info/PKG-INFO

@@ -0,0 +1,62 @@
+Metadata-Version: 2.4
+Name: keep_gpu
+Version: 0.1.0.post2
+Summary: Keep GPU is a simple cli app that keep your gpus running
+Author-email: Siyuan Wang <sywang0227@gmail.com>
+Maintainer-email: Siyuan Wang <sywang0227@gmail.com>
+License: MIT license
+Project-URL: bugs, https://github.com/Wangmerlyn/keep_gpu/issues
+Project-URL: changelog, https://github.com/Wangmerlyn/keep_gpu/blob/master/changelog.md
+Project-URL: homepage, https://github.com/Wangmerlyn/keep_gpu
+Description-Content-Type: text/x-rst
+License-File: LICENSE
+License-File: AUTHORS.rst
+Requires-Dist: typer
+Provides-Extra: dev
+Requires-Dist: coverage; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
+Dynamic: license-file
+
+========
+Keep GPU
+========
+
+
+.. image:: https://img.shields.io/pypi/v/keep_gpu.svg
+        :target: https://pypi.python.org/pypi/keep_gpu
+
+.. image:: https://img.shields.io/travis/Wangmerlyn/KeepGPU.svg
+        :target: https://travis-ci.com/Wangmerlyn/KeepGPU
+
+.. image:: https://readthedocs.org/projects/keep-gpu/badge/?version=latest
+        :target: https://keep-gpu.readthedocs.io/en/latest/?version=latest
+        :alt: Documentation Status
+
+
+.. image:: https://pyup.io/repos/github/Wangmerlyn/keep_gpu/shield.svg
+     :target: https://pyup.io/repos/github/Wangmerlyn/keep_gpu/
+     :alt: Updates
+
+
+
+Keep GPU is a simple cli app that keep your gpus running
+
+
+* Free software: MIT license
+* Documentation: https://keep-gpu.readthedocs.io.
+
+
+Features
+--------
+
+* TODO
+
+Credits
+-------
+
+This package was created with Cookiecutter_ and the `audreyr/cookiecutter-pypackage`_ project template.
+
+.. _Cookiecutter: https://github.com/audreyr/cookiecutter
+.. _`audreyr/cookiecutter-pypackage`: https://github.com/audreyr/cookiecutter-pypackage

keep_gpu-0.1.0.post2/src/keep_gpu.egg-info/SOURCES.txt

@@ -0,0 +1,29 @@
+AUTHORS.rst
+CONTRIBUTING.rst
+HISTORY.rst
+LICENSE
+MANIFEST.in
+README.rst
+pyproject.toml
+docs/Makefile
+docs/authors.rst
+docs/conf.py
+docs/contributing.rst
+docs/history.rst
+docs/index.rst
+docs/installation.rst
+docs/make.bat
+docs/readme.rst
+docs/usage.rst
+src/keep_gpu/__init__.py
+src/keep_gpu/benchmark.py
+src/keep_gpu/cli.py
+src/keep_gpu/keep_gpu.py
+src/keep_gpu.egg-info/PKG-INFO
+src/keep_gpu.egg-info/SOURCES.txt
+src/keep_gpu.egg-info/dependency_links.txt
+src/keep_gpu.egg-info/entry_points.txt
+src/keep_gpu.egg-info/requires.txt
+src/keep_gpu.egg-info/top_level.txt
+tests/__init__.py
+tests/test_keep_gpu.py

keep_gpu-0.1.0.post2/src/keep_gpu.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

keep_gpu-0.1.0.post2/src/keep_gpu.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+keep-gpu = keep_gpu.cli:main

keep_gpu-0.1.0.post2/src/keep_gpu.egg-info/requires.txt

@@ -0,0 +1,7 @@
+typer
+
+[dev]
+coverage
+mypy
+pytest
+ruff

keep_gpu-0.1.0.post2/src/keep_gpu.egg-info/top_level.txt

@@ -0,0 +1 @@
+keep_gpu

keep_gpu-0.1.0.post2/tests/__init__.py

@@ -0,0 +1 @@
+"""Unit test package for keep_gpu."""

keep_gpu-0.1.0.post2/tests/test_keep_gpu.py

@@ -0,0 +1,21 @@
+#!/usr/bin/env python
+
+"""Tests for `keep_gpu` package."""
+
+
+import unittest
+
+from keep_gpu import keep_gpu
+
+
+class TestKeep_gpu(unittest.TestCase):
+    """Tests for `keep_gpu` package."""
+
+    def setUp(self):
+        """Set up test fixtures, if any."""
+
+    def tearDown(self):
+        """Tear down test fixtures, if any."""
+
+    def test_000_something(self):
+        """Test something."""