--- a
+++ b/custom_extensions/nms/setup.py
@@ -0,0 +1,25 @@
+"""
+Created at 07.11.19 19:12
+@author: gregor
+
+"""
+
+import os, sys, site
+from pathlib import Path
+
+# recognise newly installed packages in path
+site.main()
+
+from setuptools import setup
+from torch.utils import cpp_extension
+
+dir_ = Path(os.path.dirname(sys.argv[0]))
+
+sources = [str(dir_/'src/nms_interface.cpp'), str(dir_/'src/nms.cu')]
+
+setup(name='nms_extension',
+      ext_modules=[cpp_extension.CUDAExtension(
+            'nms_extension', sources
+      )],
+      cmdclass={'build_ext': cpp_extension.BuildExtension}
+      )
\ No newline at end of file