|
libraries = sorted(glob.glob(os.path.join(path, 'libcudnn*'))) |
|
if libraries: |
|
CUDNN_LIBRARY = libraries[0] |
|
CUDNN_LIB_DIR = path |
|
break |
If there was two libcudnn files (libcudnn.5.dylib and libcudnn.7.dylib) in the folder wouldn't it result in taking the older version (libcudnn.5.dylib)?
Maybe adding a warning or something to insure its the right lib?
pytorch/tools/setup_helpers/cudnn.py
Lines 59 to 63 in 94f439c
If there was two libcudnn files (libcudnn.5.dylib and libcudnn.7.dylib) in the folder wouldn't it result in taking the older version (libcudnn.5.dylib)?
Maybe adding a warning or something to insure its the right lib?