fix: adding cuda search path to docker build.

TabbyML · Oct 28, 2023 · 6dd12ce · 6dd12ce
1 parent 2d94863
commit 6dd12ce
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/crates/llama-cpp-bindings/build.rs b/crates/llama-cpp-bindings/build.rs
@@ -14,7 +14,10 @@ fn main() {
     }
     if cfg!(feature = "cuda") {
         config.define("LLAMA_CUBLAS", "ON");
+        println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64");
+        println!("cargo:rustc-link-search=native=/opt/cuda/lib64");
         println!("cargo:rustc-link-lib=cudart");
+        println!("cargo:rustc-link-lib=culibos");
         println!("cargo:rustc-link-lib=cublas");
         println!("cargo:rustc-link-lib=cublasLt");
     }