mirror of https://github.com/hwchase17/langchain
check if libcublas is available before running extended tests
parent
248db75cd6
commit
7185fdc990
@ -0,0 +1,8 @@
|
|||||||
|
import ctypes
|
||||||
|
|
||||||
|
def is_libcublas_available():
|
||||||
|
try:
|
||||||
|
ctypes.CDLL("libcublas.so")
|
||||||
|
return True
|
||||||
|
except OSError:
|
||||||
|
return False
|
Loading…
Reference in New Issue