Merge pull request #1475 from pierotofy/rolling

Fix SIFT GPU on Windows
pull/1476/head v2.8.6
Piero Toffanin 2022-06-19 17:02:10 -04:00 zatwierdzone przez GitHub
commit 157a2eb90a
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
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)