import imp
import os
from setuptools import setup

ver = imp.load_source('version',
                      os.path.join(os.path.dirname(__file__), 'hessianfree',
                                   'version.py')).__version__

with open("README.rst") as f:
    long_description = f.read()

setup(
    name='hessianfree',
    packages=['hessianfree'],
    version=ver,
    description='Hessian-free optimization for deep networks',
    long_description=long_description,
    author='Daniel Rasmussen',
    author_email='drasmussen@princeton.edu',
    url='https://github.com/drasmuss/hessianfree',
    download_url='https://github.com/drasmuss/hessianfree/tarball/%s' % ver,
    keywords=['neural network', 'hessian free', 'deep learning'],
    license="BSD",
    classifiers=['Development Status :: 4 - Beta',
                 'Intended Audience :: Science/Research',
                 'License :: OSI Approved :: BSD License',
                 'Operating System :: Microsoft :: Windows',
                 'Operating System :: Unix',
                 'Programming Language :: Python',
                 'Programming Language :: Python :: 2.7',
                 'Topic :: Scientific/Engineering'],
)

@@ -42,16 +42,31 @@
 rt setup
+, find_packages
 %0A%0Aver = 
@@ -343,23 +343,56 @@
 ges=
-%5B'hessianfree
+find_packages(),%0A    package_data=%7B'': %5B'*.cu
 '%5D
+%7D
 ,%0A  

Include GPU code in installation