Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
bitsandbytes
Commits
2f0f0e5d
Commit
2f0f0e5d
authored
Jul 15, 2023
by
ihsanturk
Browse files
get_cuda_lib_handle brought back so import works
parent
6ec4f0c3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
0 deletions
+10
-0
bitsandbytes/cuda_setup/main.py
bitsandbytes/cuda_setup/main.py
+10
-0
No files found.
bitsandbytes/cuda_setup/main.py
View file @
2f0f0e5d
...
@@ -360,3 +360,13 @@ def evaluate_cuda_setup():
...
@@ -360,3 +360,13 @@ def evaluate_cuda_setup():
binary_name
=
f
"libbitsandbytes_cuda
{
cuda_version_string
}
_nocublaslt.so"
binary_name
=
f
"libbitsandbytes_cuda
{
cuda_version_string
}
_nocublaslt.so"
return
binary_name
,
cudart_path
,
cc
,
cuda_version_string
return
binary_name
,
cudart_path
,
cc
,
cuda_version_string
def
get_cuda_lib_handle
():
# 1. find libcuda.so library (GPU driver) (/usr/lib)
try
:
cuda
=
ct
.
CDLL
(
"libcuda.so"
)
except
OSError
:
CUDASetup
.
get_instance
().
add_log_entry
(
'CUDA SETUP: WARNING! libcuda.so not found! Do you have a CUDA driver installed? If you are on a cluster, make sure you are on a CUDA machine!'
)
return
None
return
cuda
\ No newline at end of file
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment