Setup.py - Minor change to cudnn locator

o2team · Sep 26, 2020 · 8861101 · 8861101
1 parent dee13ab
commit 8861101
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -545,6 +545,7 @@ def cudnn_checkfiles_linux():
             return list()
         cudnn_vers = chk[0]
         cudnn_path = chk[chk.find("=>") + 3:chk.find("libcudnn") - 1]
+        cudnn_path = os.path.realpath(cudnn_path)
         cudnn_path = cudnn_path.replace("lib", "include")
         cudnn_checkfiles = [os.path.join(cudnn_path, "cudnn_v{}.h".format(cudnn_vers)),
                             os.path.join(cudnn_path, "cudnn.h")]