Commit a6c78775 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Attempt to fix Dockerfile kernel compilation

parent f3738700
...@@ -55,7 +55,12 @@ def get_nvidia_cc(): ...@@ -55,7 +55,12 @@ def get_nvidia_cc():
""" """
CUDA_SUCCESS = 0 CUDA_SUCCESS = 0
libnames = ('libcuda.so', 'libcuda.dylib', 'cuda.dll') libnames = [
'libcuda.so',
'libcuda.dylib',
'cuda.dll',
'/usr/local/cuda/compat/libcuda.so',
]
for libname in libnames: for libname in libnames:
try: try:
cuda = ctypes.CDLL(libname) cuda = ctypes.CDLL(libname)
...@@ -64,7 +69,8 @@ def get_nvidia_cc(): ...@@ -64,7 +69,8 @@ def get_nvidia_cc():
else: else:
break break
else: else:
return None, "Could not load any of: " + ' '.join(libnames) raise OSError("Could not load CUDA library")
#return None, "Could not load any of: " + ' '.join(libnames)
nGpus = ctypes.c_int() nGpus = ctypes.c_int()
cc_major = ctypes.c_int() cc_major = ctypes.c_int()
......
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