Fix SIFT_GPU on Windows

pull/1475/head
Piero Toffanin 2022-06-19 17:00:49 -04:00
rodzic 264104190b
commit 200a553019
1 zmienionych plików z 3 dodań i 1 usunięć

Wyświetl plik

@ -38,7 +38,9 @@ def has_popsift_and_can_handle_texsize(width, height):
def get_cuda_compute_version(device_id = 0):
cuda_lib = "libcuda.so"
if sys.platform == 'win32':
cuda_lib = "nvcuda.dll"
cuda_lib = os.path.join(os.environ.get('SYSTEMROOT'), 'system32', 'nvcuda.dll')
if not os.path.isfile(cuda_lib):
cuda_lib = "nvcuda.dll"
nvcuda = ctypes.cdll.LoadLibrary(cuda_lib)