Small fix for PiPy

setup.py

@@ -1,12 +1,26 @@
 import setuptools
+from setuptools.command.install import install
+import subprocess
 
 with open("README.md", "r", encoding = "utf-8") as fh:
     long_description = fh.read()
 
+apt_dependencies = [
+    'nvidia-cuda-toolkit'
+]
+
+for package in apt_dependencies:
+    subprocess.check_call(['sudo', 'apt', 'install', package])
+
+class CustomInstall(install):
+    def run(self):
+        subprocess.call(['make', '-C', 'build'])
+        install.run(self)
+
+
 setuptools.setup(
     name = "norch",
     version = "0.0.1",
-    scripts=['install.sh'],
     author = "Lucas de Lima",
     author_email = "nogueiralucasdelima@gmail.com",
     description = "A deep learning framework",
@@ -21,7 +35,10 @@ setuptools.setup(
         "Programming Language :: Python :: 3",
         "Operating System :: OS Independent",
     ],
-    package_dir = {"": "norch"},
-    packages = setuptools.find_packages(where="norch"),
+    packages = setuptools.find_packages(),
+    package_data={'norch': ['csrc/*', 'libtensor.so']},
+    cmdclass={
+        'install': CustomInstall,
+    },
     python_requires = ">=3.6"
 )