Adds paths for the CUDA static library based on CUDA_HOME (#608).

Author: Andy-Jost

cuda_bindings/setup.py

@@ -31,6 +31,7 @@
     raise RuntimeError("Environment variable CUDA_HOME or CUDA_PATH is not set")
 
 CUDA_HOME = CUDA_HOME.split(os.pathsep)
+
 if os.environ.get("PARALLEL_LEVEL") is not None:
     warn(
         "Environment variable PARALLEL_LEVEL is deprecated. Use CUDA_PYTHON_PARALLEL_LEVEL instead",
@@ -238,6 +239,8 @@ def generate_output(infile, local):
     os.path.dirname(sysconfig.get_path("include")),
 ] + include_path_list
 library_dirs = [sysconfig.get_path("platlib"), os.path.join(os.sys.prefix, "lib")]
+cudalib_subdir = r"lib\x64" if sys.platform == "win32" else "lib64"
+library_dirs.extend(os.path.join(prefix, cudalib_subdir) for prefix in CUDA_HOME)
 
 extra_compile_args = []
 extra_cythonize_kwargs = {}