diff --git a/dnnlib/tflib/custom_ops.py b/dnnlib/tflib/custom_ops.py index ed31b7691546b844195321283fe52129d6ce9525..b97691c17e46245152d41c0d827f3f64d8bb7706 100755 --- a/dnnlib/tflib/custom_ops.py +++ b/dnnlib/tflib/custom_ops.py @@ -69,7 +69,7 @@ def _run_cmd(cmd): raise RuntimeError('NVCC returned an error. See below for full command line and output log:\n\n%s\n\n%s' % (cmd, output)) def _prepare_nvcc_cli(opts): - cmd = 'nvcc ' + opts.strip() + cmd = 'nvcc --std=c++11 -DNDEBUG ' + opts.strip() cmd += ' --disable-warnings' cmd += ' --include-path "%s"' % tf.sysconfig.get_include() cmd += ' --include-path "%s"' % os.path.join(tf.sysconfig.get_include(), 'external', 'protobuf_archive', 'src')