1 files changed, 5 insertions, 0 deletions
diff --git a/wscript b/wscript
index e30727c8c..439fc4fa5 100644
--- a/wscript
+++ b/wscript
@@ -78,6 +78,7 @@ def options(opt):
     opt.add_option('--enable-disk',       action='store_true', default=False, help='build dcpomatic2_disk tool; requires Boost process, lwext4 and nanomsg libraries')
     opt.add_option('--warnings-are-errors', action='store_true', default=False, help='build with -Werror')
     opt.add_option('--wx-config',         help='path to wx-config')
+    opt.add_option('--cuda-path',         help='path to directory containing include/cuda_runtime_api.h etc.', default='/usr/local/cuda')
 
 def configure(conf):
     conf.load('compiler_cxx')
@@ -147,6 +148,8 @@ def configure(conf):
         except conf.errors.ConfigurationError:
             pass
 
+    conf.env.append_value('INCLUDES', os.path.join(conf.options.cuda_path, 'include'))
+
     #
     # Windows/Linux/OS X specific
     #
@@ -558,6 +561,8 @@ def configure(conf):
                            lib=deps,
                            uselib_store='BOOST_PROCESS')
 
+    conf.check(header_name='nvjpeg2k.h', mandatory=False)
+
     # Other stuff
 
     conf.find_program('msgfmt', var='MSGFMT')