diff --git a/crates/llama-cpp-bindings/build.rs b/crates/llama-cpp-bindings/build.rs index a0c511b..59a2106 100644 --- 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"); }