Fix: add Debian's installation to default paths (#72)

When you install `nvidia-cuda-toolkit` on Debian, it installs the CUDA libraries into `/usr/lib/cuda`. Add this path to the default search paths.
Rust-GPU · May 17, 2022 · b8127c7 · b8127c7
1 parent 2de8180
commit b8127c7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/crates/find_cuda_helper/src/lib.rs b/crates/find_cuda_helper/src/lib.rs
@@ -130,6 +130,7 @@ pub fn find_cuda_lib_dirs() -> Vec<PathBuf> {
     for e in glob::glob("/usr/local/cuda-*").unwrap().flatten() {
         candidates.push(e)
     }
+    candidates.push(PathBuf::from("/usr/lib/cuda"));
 
     let mut valid_paths = vec![];
     for base in &candidates {