nvcc の include ディレクトリの階層が変わっているので CUDA_TOOLKIT_INCLUDE を設定する

shiguredo · Jan 18, 2024 · b1e76b2 · b1e76b2
1 parent bf9233c
commit b1e76b2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/run.py b/run.py
@@ -1713,6 +1713,7 @@ def main():
             cmake_args.append('-DUSE_NVCODEC_ENCODER=ON')
             if platform.target.os == 'windows':
                 cmake_args.append(f"-DCUDA_TOOLKIT_ROOT_DIR={cmake_path(os.path.join(install_dir, 'cuda', 'nvcc'))}")
+                cmake_args.append(f"-DCUDA_TOOLKIT_INCLUDE={cmake_path(os.path.join(install_dir, 'cuda', 'nvcc', 'crt'))}")
 
         if platform.target.os in ('windows', 'ubuntu') and platform.target.arch == 'x86_64':
             cmake_args.append('-DUSE_VPL_ENCODER=ON')