From 88611018b32dc7f5ea8b19319457b4d85639b81b Mon Sep 17 00:00:00 2001 From: torzdf <36920800+torzdf@users.noreply.github.com> Date: Sun, 27 Sep 2020 00:56:39 +0100 Subject: [PATCH] Setup.py - Minor change to cudnn locator --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index b68875b422..97db789bf6 100755 --- 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")]