Unverified Commit bd380048 authored by Tim Geypens's avatar Tim Geypens Committed by GitHub
Browse files

fall back after failing ldconfig-based lib loading for cuDNN (#2277)


Signed-off-by: default avatarTim Geypens <tim.geypens@gmail.com>
parent 05dc1e62
......@@ -252,9 +252,7 @@ def _load_cudnn():
return handle
# Attempt to locate libcudnn via ldconfig
libs = subprocess.check_output(
f"ldconfig -p | grep 'libcudnn{_get_sys_extension()}'", shell=True
)
libs = subprocess.check_output(["ldconfig", "-p"])
libs = libs.decode("utf-8").split("\n")
sos = []
for lib in libs:
......@@ -284,9 +282,7 @@ def _load_nvrtc():
return handle
# Attempt to locate NVRTC via ldconfig
libs = subprocess.check_output(
f"ldconfig -p | grep 'libnvrtc{_get_sys_extension()}'", shell=True
)
libs = subprocess.check_output(["ldconfig", "-p"])
libs = libs.decode("utf-8").split("\n")
sos = []
for lib in libs:
......@@ -316,9 +312,7 @@ def _load_curand():
return handle
# Attempt to locate cuRAND via ldconfig
libs = subprocess.check_output(
f"ldconfig -p | grep 'libcurand{_get_sys_extension()}'", shell=True
)
libs = subprocess.check_output(["ldconfig", "-p"])
libs = libs.decode("utf-8").split("\n")
sos = []
for lib in libs:
......
Markdown is supported
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