ZhWL123456

Add a line in the 'set_up.py': sources = [os.path.join(extensions_dir, s) for s in sources] include_dirs = [extensions_dir] include_dirs.append('/usr/local/cuda/include') #I added this line ext_modules = [ extension( "_ext", sources, include_dirs=include_dirs, define_macros=define_macros,...

how do you finish it? I face the same problem of Segmentation fault. And I dont know how to solve it! Thank u very much!