File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ( ) ;
You can’t perform that action at this time.
0 commit comments