如何檢查 opencv 是否使用 GPU？

Question

我需要知道當前的 opencv 安裝是否使用 GPU。 我試過print(cv2.getBuildInformation())但這不是我想要的。 我也試過getCudaEnabledDeviceCount()這不起作用並且也會引發錯誤。

Answer 1

如果您已經安裝了cuda ，那么您現在可以使用 opencv 中的內置function 。

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

count返回已安裝的支持 CUDA 的設備的數量。

您可以使用此 function 來處理所有情況。

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測試