Как проверить, использует ли opencv графический процессор или нет?

Question

Мне нужно знать, использует ли текущая установка opencv графический процессор или нет. Я пытался print(cv2.getBuildInformation()), но это не то, что я ищу. Я также попробовал getCudaEnabledDeviceCount(), это не работает и тоже выдает ошибку.

Answer 1

Если вы установили cuda , в opencv есть встроенная функция, которую вы можете использовать сейчас.

import cv2
count = cv2.cuda.getCudaEnabledDeviceCount()
print(count)

count возвращает число установленных устройств с поддержкой CUDA.

Вы можете использовать эту функцию для обработки всех случаев.

def is_cuda_cv(): # 1 == using cuda, 0 = not using cuda
    try:
        count = cv2.cuda.getCudaEnabledDeviceCount()
        if count > 0:
            return 1
        else:
            return 0
    except:
        return 0

Протестировано с opencv 4.2.0