-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
48 lines (43 loc) · 1.24 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#!/usr/bin/env python
import os
import re
from setuptools import setup, find_packages
version_re = r'^__version__ = ["\']([^"\']*)["\']$'
author_re = r'^__author__ = ["\'](.*)["\']$'
INIT = open(os.path.join('gpucrate', '__init__.py')).read()
version = re.search(version_re, INIT, re.M).group(1)
author = re.search(author_re, INIT, re.M).group(1)
README = open('README.md').read()
setup(
name='gpucrate',
version=version,
packages=find_packages(),
author=author,
url="https://github.com/jtriley/gpucrate",
description=("gpucrate creates hard-linked GPU driver volumes "
"for use with docker, singularity, etc."),
long_description=README,
install_requires=[
"sh>=1.11",
"nvidia-ml-py>=7.352.0",
"PyYAML>=3.11",
],
setup_requires=[
'pytest-runner>=2.9'
],
tests_require=[
"pytest>=3.0.3",
"pytest-cov>=2.4.0",
"pytest-flake8>=0.7",
"testfixtures>=4.10.0",
"mock>=2.0.0",
],
entry_points=dict(console_scripts=[
'gpucrate = gpucrate.cli:main',
'singularity-gpu = gpucrate.cli:singularity_gpu',
]),
include_package_data=True,
package_data={
'gpucrate.tests': ['data/*.txt'],
},
)