【问题标题】:How to check if opencv is using GPU or not?如何检查opencv是否使用GPU?
【发布时间】:2020-08-12 23:45:13
【问题描述】:

我需要知道当前的 opencv 安装是否使用 GPU。我试过print(cv2.getBuildInformation()) 但这不是我想要的。我也试过getCudaEnabledDeviceCount()这不起作用,也会抛出错误。

【问题讨论】:

    标签: python opencv gpu


    【解决方案1】:

    如果你已经安装了cudaopencv中有一个内置函数,你现在就可以使用了。

    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

    【讨论】:

    • 请不要发布文本屏幕截图,发布文本本身(复制/粘贴等)
    • 我已经发布了相关代码,截图只是为了演示我的opencv版本,因为在较低版本上尝试这个可能不起作用。我应该删除它吗?
    • 如果你能够复制/粘贴内容而不是屏幕截图——图片必须去=)
    猜你喜欢
    • 2022-01-27
    • 2019-01-12
    • 2018-06-17
    • 1970-01-01
    • 1970-01-01
    • 2018-11-23
    • 2017-11-16
    • 2020-07-06
    相关资源
    最近更新 更多