Skip to content
Snippets Groups Projects
Commit 2237d7c6 authored by patmjen's avatar patmjen
Browse files

Improve search for cuDNN in init script 

Some cuDNN modules on the system are just listed as CUDA_MAJOR_VERSION.X. Thus, if no cuDNN module is found for the exact CUDA version, we then look for a .X module.
parent 7fc5c520
No related branches found
No related tags found
Hide whitespace changes
Inline Side-by-side
Showing
with 6 additions and 0 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment