Merge branch 'release/1.1.0'

Related with #11 and #9

Author: devmessias

emate/__init__.py

@@ -7,10 +7,10 @@
 try:
     import cupy as cp
 except:
-    print("Warning: Cupy package not found")
+    print("Warning: CUPy package not found")
 
-__version__ = "1.0.4"
-__license__ = ""
+__version__ = "1.1.0"
+__license__ = "MIT"
 __author__ = "Bruno Messias; Thomas K Peron"
 __author_email__ = "[email protected]"
 __name__ = "eMaTe"

requiriments_cpu.txt

@@ -0,0 +1,3 @@
+numpy==1.19.0
+scipy==1.5.0
+tensorflow==1.15.3

requiriments_gpu.txt

@@ -0,0 +1,3 @@
+numpy==1.19.0
+scipy==1.5.0
+tensorflow-gpu==1.15.2

setup.py

@@ -1,19 +1,73 @@
 from setuptools import setup, find_packages
+import ctypes
 
 
+def checkCUDAisAvailable():
+    """
+    This function check if any of this possible libs are available.
+    see https://gist.github.com/f0k/63a664160d016a491b2cbea15913d549
+
+    Returns:
+    --------
+        libsOk : bool
+            If True then CUDA is available
+    """
+    # some possible lib names 
+    libnames = ('libcuda.so', 'libcuda.dylib', 'cuda.dll')
+    libsOk = True
+    for libname in libnames:
+        try:
+            cuda = ctypes.CDLL(libname)
+        except OSError:
+            continue
+        else:
+            break
+    else:
+        libsOk = False
+    return libsOk
+
+def getRequirements():
+    """
+    This function it's used in order to get the package names. Which
+    depends on the libs available in the machine. 
+
+    Return:
+    -------
+        conditionalRequirements: list
+            A list of strings containing the pip pkgs.
+    """
+    cudaLibsOk = checkCUDAisAvailable()   
+    
+    conditionalRequirements = []
+    if cudaLibsOk:
+        conditionalRequirements += ["tensorflow-gpu==1.15.3", ]
+    else:
+        print("\n CUDA it's not available in your machine.")
+        print(" You won't be able to use the GPU support.\n")
+        conditionalRequirements += ["tensorflow==1.15.3", ]
+
+    return conditionalRequirements
+
+conditionalRequirements = getRequirements()
+install_requires = ["scipy", "numpy"] + conditionalRequirements
+
 with open("README.md", "r") as f:
     README_TEXT = f.read()
 
 setup(
     name="emate",
-    version="v1.0.4",
+    version="v1.1.0",
     packages=find_packages(exclude=["build", ]),
     long_description=README_TEXT,
     long_description_content_type="text/markdown",
-    install_requires=["tensorflow==1.15.0", "scipy", "numpy"],
+    install_requires=install_requires,
     include_package_data=True,
     license="MIT",
-    description="",
+    description="""eMaTe can run in both CPU and GPU and can 
+        estimate the spectral density and related trace functions, 
+        such as entropy and Estrada index, even in matrices 
+        (directed or undirected graphs) with 
+        million of nodes.""",
     author_email="[email protected]",
     author="Bruno Messias; Thomas K Peron",
     download_url="https://github.com/stdogpkg/emate/archive/v1.0.4.tar.gz",