在 Windows 10 上使用 Cuda 構建 Python OpenCV 時遇到問題

Question

我正在嘗試在 Windows 10 上構建支持 Python 3.9.5 的 Cuda 的 OpenCV 3.4.15 並且卡住了。 我已經學習了一些教程，我認為我已經接近了，但我似乎遺漏了一些東西。

我已經在啟用 WITH_CUDA 的情況下運行 CMake，一切似乎都運行良好。 我打開 OpenCV.sln 並構建 ALL_BUILD，然后構建 INSTALL。 一切都成功構建，C++ 端的 Cuda 支持運行良好。 在 Python 方面，我最終將文件添加到以下路徑，Python 中的非 Cuda OpenCV 似乎工作正常。

C:\\Users\\Name\\AppData\\Local\\Programs\\Python\\Python39\\Lib\\site-packages\\cv2

如果我運行cv2.getBuildInformation()我會看到以下內容，這讓我覺得我成功了（這在 OpenCV 的 pip 安裝版本中沒有出現）。

NVIDIA CUDA:                   YES (ver 11.4, CUFFT CUBLAS)
  NVIDIA GPU arch:             35 37 50 52 60 61 70 75 80 86
  NVIDIA PTX archs:

但是，如果我運行 cv2.cuda.getCudaEnabledDeviceCount() 我得到以下信息。

AttributeError: module 'cv2.cuda' has no attribute 'getCudaEnabledDeviceCount'

當我使用通過 pip 安裝的 OpenCV 運行相同的命令時，這成功輸出了 0。

我注意到的一件事是，我從源代碼創建的 site-packages/cv2 目錄只有大約 6MB，而我從 pip install 獲得的 site-packages/cv2 目錄接近 100MB，所以似乎有些不對勁。

Answer 1

似乎構建 OpenCV 4.5.4 而不是 OpenCV 3.4.15 解決了這個問題。 現在可以使用與 OpenCV Cuda 實現的 Python 綁定。