Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 5d77c102 authored by hhakim's avatar hhakim
Browse files

Set priority of CUDA versions in gpu_mod lib loading.

parent f218a10c
Branches
Tags 3.38.9
No related merge requests found
Pipeline #834172 skipped
...@@ -88,9 +88,9 @@ def enable_gpu_mod(libpaths=None, backend='cuda', silent=False, fatal=False): ...@@ -88,9 +88,9 @@ def enable_gpu_mod(libpaths=None, backend='cuda', silent=False, fatal=False):
# don't use os.path.join or os.path.sep because anyway # don't use os.path.join or os.path.sep because anyway
# lib name suffix and prefix depend on OS # lib name suffix and prefix depend on OS
libpaths = [] libpaths = []
for gpu_backend in ['-cu9.2', '-cu11.4', '']: # the last one is for gpu_backend in ['-cu12.1', '-cu11.4', '']: # the last one is
# for descendant/backward compatibility # for unversioned lib
# (the backend are sorted per order of preference) # (the versions are sorted per order of preference)
if sys.platform == 'linux': if sys.platform == 'linux':
libpaths += [pyfaust_path+"/lib/libgm"+gpu_backend+".so"] libpaths += [pyfaust_path+"/lib/libgm"+gpu_backend+".so"]
elif sys.platform == 'darwin': elif sys.platform == 'darwin':
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment