def cuda_and_linux():
return has_cuda() and platform.system() == "Linux"
from poly import has_cuda
if has_cuda():
try:
from cuGraph import *
except ImportError:
from networkx import *
else:
from networkx import *
except ImportError:
import ez_setup
ez_setup.use_setuptools()
from setuptools import setup, find_packages
from poly import has_cuda
import platform
def cuda_and_linux():
return has_cuda() and platform.system() == "Linux"
requirements = [
"tensorflow-gpu" if has_cuda() else "tensorflow", "numba", "spacy",
"keras", "sklearn", "networkx", "pandas", "numpy"
]
if has_cuda():
requirements = requirements + ["pyCuda"]
if cuda_and_linux():
requirements = requirements + ["cudf", "cuml", "cugraph"]
elif platform.system() == "Windows":
pass
elif platform.system() == "Darwin":
requirements = requirements + ["plaidml", "plaidml-keras"]
setup(
name="poly",