nvitop/setup.py
2021-06-11 22:59:04 +08:00

67 lines
2.3 KiB
Python

"""An interactive NVIDIA-GPU process viewer, the one-stop solution for GPU process management."""
import platform
from setuptools import find_packages, setup
__version__ = '0.3.3'
__license__ = 'GPLv3'
__author__ = __maintainer__ = 'Xuehai Pan'
__email__ = 'XuehaiPan@pku.edu.cn'
extra_description = {}
try:
with open('README.md', mode='r') as doc:
extra_description['long_description'] = doc.read()
extra_description['long_description_content_type'] = 'text/markdown'
except (OSError, UnicodeError):
pass
setup(
name='nvitop',
version=__version__,
description=__doc__,
**extra_description,
license=__license__,
author=__author__,
author_email=__email__,
url="https://github.com/XuehaiPan/nvitop",
packages=find_packages(include=['nvitop', 'nvitop.*']),
entry_points={'console_scripts': ['nvitop=nvitop.cli:main']},
install_requires=(['windows-curses'] if platform.system() == 'Windows' else []) + [
'nvidia-ml-py == 11.450.51',
'psutil',
'cachetools',
'termcolor',
],
python_requires='>=3.5',
classifiers=[
'Development Status :: 4 - Beta',
'License :: OSI Approved :: GNU General Public License v3 (GPLv3)',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'Programming Language :: Python :: 3.10',
'Operating System :: Microsoft :: Windows',
'Operating System :: POSIX :: Linux',
'Environment :: GPU',
'Environment :: GPU :: NVIDIA CUDA',
'Environment :: Console',
'Environment :: Console :: Curses',
'Intended Audience :: Developers',
'Intended Audience :: End Users/Desktop',
'Intended Audience :: System Administrators',
'Topic :: System :: Hardware',
'Topic :: System :: Monitoring',
'Topic :: System :: Systems Administration',
'Topic :: Utilities',
],
keywords='nvidia, nvidia-smi, NVIDIA, NVML, CUDA, GPU, top, monitoring',
project_urls={
'Bug Reports': 'https://github.com/XuehaiPan/nvitop/issues',
'Source': 'https://github.com/XuehaiPan/nvitop',
},
)