root
/
tabby
mirror of https://gitee.com/zhang_1334717033/tabby.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: adding cuda search path to docker build.

release-notes-05
Meng Zhang 2023-10-27 19:40:35 -07:00 committed by GitHub
parent 2d948639be
commit 6dd12ce1ec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
crates/llama-cpp-bindings/build.rs
View File

@ -14,7 +14,10 @@ fn main() {
} }
if cfg!(feature = "cuda") { if cfg!(feature = "cuda") {
config.define("LLAMA_CUBLAS", "ON"); 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=cudart");
println!("cargo:rustc-link-lib=culibos");
println!("cargo:rustc-link-lib=cublas"); println!("cargo:rustc-link-lib=cublas");
println!("cargo:rustc-link-lib=cublasLt"); println!("cargo:rustc-link-lib=cublasLt");
} }