"tests/data_utils.py" did not exist on "0333b7a3b7ffdfbf48db6290468198a238e31a8b"
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():
"""
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:
try:
cuda = ctypes.CDLL(libname)
......@@ -64,7 +69,8 @@ def get_nvidia_cc():
else:
break
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()
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