We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 076ba79 commit b27d5f2Copy full SHA for b27d5f2
crates/find_cuda_helper/src/lib.rs
@@ -107,6 +107,21 @@ pub fn find_cuda_lib_dirs() -> Vec<PathBuf> {
107
vec![]
108
}
109
110
+pub fn read_env() -> Vec<PathBuf> {
111
+ if let Ok(path) = env::var("CUDA_LIBRARY_PATH") {
112
+ // The location of the libcuda, libcudart, and libcublas can be hardcoded with the
113
+ // CUDA_LIBRARY_PATH environment variable.
114
+ let split_char = if cfg!(target_os = "windows") {
115
+ ";"
116
+ } else {
117
+ ":"
118
+ };
119
+ path.split(split_char).map(PathBuf::from).collect()
120
121
+ vec![]
122
+ }
123
+}
124
+
125
#[cfg(not(target_os = "windows"))]
126
pub fn find_cuda_lib_dirs() -> Vec<PathBuf> {
127
let mut candidates = read_env();
0 commit comments