機器學習:如何編譯OpenCV以包括CUDA GPU支持?
本文將逐步介紹如何編譯OpenCV以包括CUDA GPU支持,以便可以在基于視覺的機器學習項目中使用它。
Pre:我之所以決定寫這篇文章,是因為我發(fā)現(xiàn)現(xiàn)有指南缺少一些更詳細的信息,無法說明如何使用CUDA GPU支持從源代碼構(gòu)建OpenCV,以便將其導(dǎo)入python3.8conda環(huán)境。大多數(shù)人都以構(gòu)建過程結(jié)束,但低估這只是使OpenCV在項目中正常工作的第一步。
先決條件
在開始之前,我們必須下載一些文件,安裝一些程序。我還將假設(shè)你正在將Anaconda軟件包管理器環(huán)境用于Python 3.8,并將其設(shè)置為啟用的默認python和環(huán)境變量。如果沒有,我不確定以下內(nèi)容是否對你有用。我不建議偏離主要任務(wù)。
Anaconda軟件包管理器環(huán)境:https://www.a(chǎn)naconda.com/distribution/#download-section
1. Visual Studio 2019
請確保VisualStudio 2019安裝了“帶有C++的桌面開發(fā)”包。這是在Visual Studio的安裝過程中完成的。如果你已經(jīng)安裝了VS2019,但是不確定是否已安裝工作負載,可以通過安裝可執(zhí)行文件重新安裝(修改)來完成。
使用C ++開發(fā)的Visual Studio 2019
2. OpenCV和OpenCV貢獻文件
在計算機上的某個位置創(chuàng)建一個名為“ OpenCV_Build”的文件夾。它不必確切地命名為這個,你要看一眼就知道它是什么。然后,你將同時下載OpenCV和OpenCV contrib.zip。
OpenCV:https://github.com/opencv/opencv/archive/4.4.0.zip
OpenCV contrib:https://github.com/opencv/opencv_contrib/archive/4.4.0.zip
下載完成后,解壓縮zip并將生成的文件夾移到OpenCV_Build中。它應(yīng)該類似于:OpenCV_Build/opencv-4.5.2OpenCV_Build/opencv_contrib-4.5.2
你要確保它們都是相同的版本(在本例中為4.5.2)。
我還想確保它們是頂級目錄,而不是嵌套在具有相同名稱的文件夾中,但這不是完全必要的,只是使得我們的環(huán)境路徑更加精簡。
3. CMake GUI
下載并安裝CMake GUI
CMake GUI:https://cmake.org/
我使用的是3.20.2版,但是安裝哪個版本都沒有關(guān)系。
4. CUDA 11.0工具包
下一步,你必須在Nvidia網(wǎng)站上注冊才能下載并安裝CUDA工具包,該工具包既快速又簡單,完全不需要花費很多時間。
下載CUDA工具包:https://developer.nvidia.com/cuda-downloads
5. cuDNN
此步驟與上一步非常相似,但是你必須下載cuDNN才能利用DNN CUDA后端。
下載cuDNN:https://developer.nvidia.com/rdp/cudnn-download
在Nvidia開發(fā)人員程序中注冊后,你將需要選擇顯示以下內(nèi)容的版本:“for CUDA 11.x”.
6.提取cuDNN文件
cuDNN下載完成后,解壓縮zip并打開結(jié)果文件夾。在內(nèi)部,你應(yīng)該看到一個名為cuda的文件夾,將其打開,然后剩下3個子文件夾,分別為**“ bin”,“ include”和“ lib”。**
你將把文件從每個子文件夾復(fù)制到 C:Program FilesNVIDIA GPU Computing ToolkitCUDA11.3 中的相應(yīng)文件夾中。
downloads/cuda/bin contents to: CUDA/v11.3/bindownloads/cuda/include contents to: CUDA/v11.3/includedownloads/cuda/lib/x64 contents to: CUDA/v11.3/lib/x64
7.下載Ninja以獲得更快的構(gòu)建時間
Ninja將極大地加快OpenCV的構(gòu)建過程。
Ninja:https://github.com/ninja-build/ninja/releases/download/v1.10.2/ninja-win.zip
我個人發(fā)現(xiàn),Ninja從1小時45分鐘縮短到不到15分鐘。解壓縮它,然后將Ninja.exe放在你之前創(chuàng)建的“ OpenCV_Build”文件夾中。將所有文件放入各自的文件夾后,應(yīng)完成prereqs部分。
建議:設(shè)置Conda Env
我建議你使用Anaconda,以確保你所做的一切與我所做的相同。你可以使用以下命令設(shè)置conda env:conda create -n py38 anaconda python=3.8
建議:為簡單起見,創(chuàng)建一些.bat文件
為簡單起見,我建議制作一些.bat文件,以防萬一在構(gòu)建過程中出現(xiàn)問題。這樣,你可以快速對cmd進行編輯,然后重新運行它,而不必經(jīng)歷一次又一次地重新設(shè)置所有環(huán)境變量的過程。當出現(xiàn)問題時,它還有助于將事情分解成更易于管理的狀態(tài),而且?guī)缀蹩隙〞鲥e。
請輸入評論內(nèi)容...
請輸入評論/評論長度6~500個字
最新活動更多
-
即日-10.29立即報名>> 2024德州儀器嵌入式技術(shù)創(chuàng)新發(fā)展研討會
-
10月31日立即下載>> 【限時免費下載】TE暖通空調(diào)系統(tǒng)高效可靠的組件解決方案
-
即日-11.13立即報名>>> 【在線會議】多物理場仿真助跑新能源汽車
-
11月14日立即報名>> 2024工程師系列—工業(yè)電子技術(shù)在線會議
-
12月19日立即報名>> 【線下會議】OFweek 2024(第九屆)物聯(lián)網(wǎng)產(chǎn)業(yè)大會
-
即日-12.26火熱報名中>> OFweek2024中國智造CIO在線峰會
推薦專題
- 1 Intel宣布40年來最重大轉(zhuǎn)型:年底前裁員15000人、拋掉2/3房產(chǎn)
- 2 因美封殺TikTok,字節(jié)股價骨折!估值僅Meta1/5
- 3 宏山激光重磅發(fā)布行業(yè)解決方案,助力智能制造產(chǎn)業(yè)新飛躍
- 4 國產(chǎn)AI芯片公司破產(chǎn)!白菜價拍賣
- 5 具身智能火了,但規(guī)模落地還需時間
- 6 國產(chǎn)英偉達們,抓緊沖刺A股
- 7 三次錯失風口!OpenAI前員工殺回AI編程賽道,老東家捧金相助
- 8 英特爾賦能智慧醫(yī)療,共創(chuàng)數(shù)字化未來
- 9 英偉達的麻煩在后頭?
- 10 將“網(wǎng)紅”變成“商品”,AI“爆改”實力拉滿
- 高級軟件工程師 廣東省/深圳市
- 自動化高級工程師 廣東省/深圳市
- 光器件研發(fā)工程師 福建省/福州市
- 銷售總監(jiān)(光器件) 北京市/海淀區(qū)
- 激光器高級銷售經(jīng)理 上海市/虹口區(qū)
- 光器件物理工程師 北京市/海淀區(qū)
- 激光研發(fā)工程師 北京市/昌平區(qū)
- 技術(shù)專家 廣東省/江門市
- 封裝工程師 北京市/海淀區(qū)
- 結(jié)構(gòu)工程師 廣東省/深圳市