mirror of
https://github.com/XuehaiPan/nvitop.git
synced 2026-05-16 06:06:13 -06:00
58 lines
1.4 KiB
Python
Executable file
58 lines
1.4 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# To install `nvitop` with specific version of `nvidia-ml-py`, use:
|
|
#
|
|
# pip install nvidia-ml-py==xx.yyy.zz nvitop
|
|
#
|
|
# or
|
|
#
|
|
# pip install 'nvitop[pynvml-xx.yyy.zz]'
|
|
#
|
|
|
|
# pylint: disable=missing-module-docstring
|
|
|
|
import pathlib
|
|
import sys
|
|
|
|
from setuptools import setup
|
|
|
|
|
|
VERSION_FILE = pathlib.Path(__file__).absolute().parent / 'nvitop' / 'version.py'
|
|
|
|
try:
|
|
from nvitop import version
|
|
except ImportError:
|
|
sys.path.insert(0, str(VERSION_FILE.parent))
|
|
import version
|
|
|
|
|
|
VERSION_CONTENT = None
|
|
if not version.__release__:
|
|
import re
|
|
|
|
VERSION_CONTENT = VERSION_FILE.read_text(encoding='UTF-8')
|
|
VERSION_FILE.write_text(
|
|
data=re.sub(
|
|
r"""__version__\s*=\s*('[^']+'|"[^"]+")""",
|
|
r"__version__ = '{}'".format(version.__version__),
|
|
string=VERSION_CONTENT,
|
|
),
|
|
encoding='UTF-8',
|
|
)
|
|
|
|
try:
|
|
setup(
|
|
name='nvitop',
|
|
version=version.__version__,
|
|
extras_require={
|
|
'cuda10': ['nvidia-ml-py == 11.450.51'],
|
|
**{
|
|
# The identifier could not start with numbers, add a prefix `pynvml-`
|
|
'pynvml-{}'.format(pynvml): ['nvidia-ml-py == {}'.format(pynvml)]
|
|
for pynvml in version.PYNVML_VERSION_CANDIDATES
|
|
},
|
|
},
|
|
)
|
|
finally:
|
|
if VERSION_CONTENT is not None:
|
|
VERSION_FILE.write_text(data=VERSION_CONTENT, encoding='UTF-8')
|