Skip to content

Commit 029c225

Browse files
CharryWuLegNeato
authored andcommitted
fix(cudnn-sys): update Windows default cuDNN include paths
Made-with: Cursor
1 parent c83adae commit 029c225

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

crates/cudnn-sys/build/cudnn_sdk.rs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,8 +73,15 @@ impl CudnnSdk {
7373
];
7474
#[cfg(target_os = "windows")]
7575
const CUDNN_DEFAULT_PATHS: &[&str] = &[
76+
// Standalone cuDNN installs following NVIDIA's documentation.
7677
"C:/Program Files/NVIDIA/CUDNN/v9.x/include",
7778
"C:/Program Files/NVIDIA/CUDNN/v8.x/include",
79+
// CUDA Toolkit installs that bundle cuDNN headers.
80+
// These are the default Windows install locations for recent CUDA versions.
81+
"C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.2/include",
82+
"C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8/include",
83+
"C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.6/include",
84+
"C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v13.0/include",
7885
];
7986

8087
let mut cudnn_paths: Vec<&Path> = CUDNN_DEFAULT_PATHS.iter().map(Path::new).collect();

0 commit comments

Comments
 (0)