[英]How to check if opencv is using GPU or not?
我需要知道当前的 opencv 安装是否使用 GPU。 我试过print(cv2.getBuildInformation())
但这不是我想要的。 我也试过getCudaEnabledDeviceCount()
这不起作用并且也会引发错误。
如果您已经安装了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
测试
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.