diff --git a/third_party/gpus/crosstool/windows/msvc_wrapper_for_nvcc.py.tpl b/third_party/gpus/crosstool/windows/msvc_wrapper_for_nvcc.py.tpl index 7e0674963bf..0d1423f8ba0 100644 --- a/third_party/gpus/crosstool/windows/msvc_wrapper_for_nvcc.py.tpl +++ b/third_party/gpus/crosstool/windows/msvc_wrapper_for_nvcc.py.tpl @@ -153,6 +153,13 @@ def InvokeNvcc(argv, log=False): ] _, argv = GetOptionValue(argv, '--no-cuda-include-ptx') + # nvcc doesn't respect the INCLUDE and LIB env vars from MSVC, + # so we explicity specify the system include paths and library search paths. + if 'INCLUDE' in os.environ: + nvccopts += [('--system-include="%s"' % p) for p in os.environ['INCLUDE'].split(";")] + if 'LIB' in os.environ: + nvccopts += [('--library-path="%s"' % p) for p in os.environ['LIB'].split(";")] + nvccopts += nvcc_compiler_options nvccopts += undefines nvccopts += defines