Win10上使用VS2015編譯Caffe2


Caffe2的官網:https://caffe2.ai/

1、下載、安裝及相關准備

在Caffe2的官網點擊"Get Started",即進入安裝說明頁面。官方還未提供編譯好的binaries,所以需要自己使用源代碼編譯。

選擇平台如下:

按照官網,需要Win10系統及以上,但是似乎有些博客也有使用了Win7,但我本人是使用Win10,所以本文只針對Win10系統展開。

在准備階段,我主要下載或安裝的有:

1、Python2.7.14 https://www.python.org/downloads/release/python-2714/ 

2、cmake 3.11 https://cmake.org/download/

3、CUDA9.0 https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

4、cuDNN v7.0.5 for CUDA9.0 https://developer.nvidia.com/rdp/cudnn-archive

5、visual stduio 2015 Enterprise(用Community和Professional應該也是沒問題的,只是剛好有人給我Enterprise的安裝包)

6、Python的庫,只安裝了必須的幾個:future、hypothesis、numpy、protobuf、six,可選的那些我都沒安裝(我是使用pip install安裝的 )

7、另外,我原來自己還有安裝了Opencv3.4,因為我是使用Caffe2來處理圖像的

8、安裝git,版本應該無所謂吧,主要是用來下載源碼

9、下載caffe2源碼, caffe2(https://github.com/caffe2/caffe2)現在已和pytorch(https://github.com/pytorch/pytorch)合並了,所以使用git下載:

git clone --recursive https://github.com/pytorch/pytorch.git

接着說一下安裝過程中的一些注意或者遇到的問題:

1、安裝vs2015一般選自定義,然后勾選Visual C++的相關選項。

2、安裝vs失敗很麻煩,因為有時候卸載不干凈會影響CUDA安裝里面Visual studio integrate組件的安裝(稍后會寫另一篇博客來說明)

3、下載caffe2源碼不要直接Download Zip下載壓縮包,那樣會導致第三方庫缺失,即third_party文件夾里面的一些文件。但是目前從github上之間git clone源碼速度很慢,都是10+k甚至幾k/s,更崩潰的是下載途中還會經常中斷,然后又得重新來了,所以我試了很多次都沒成功,翻牆速度也很慢(也可能是我的VPN不給力),后來是托人在香港那邊下載的。

4、下載cuDNN解壓縮后,是一個包含include、lib和bin的文件夾,將這三個文件夾里的文件分別拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\文件夾下的include、lib和bin下(當然,以你的CUDA的安裝路徑為准,我是采用默認安裝路徑

2、編譯安裝

git得到pytorch文件夾后,在pytorch的scripts文件夾里有build_windows.bat腳本文件,可以用記事本、Notepad或者sublime text等編輯器打開,如果懂得腳本和Cmake命令就可以完全讀懂。里面主要需要做的是指定編譯器和開啟編譯選項。在該文件里加入這兩句,可以在“pip install pyyaml”語句上方:

set USE_CUDA=ON
set CMAKE_GENERATOR="Visual Studio 14 2015 Win64"

CMAKE_GENERATOR即是設置編譯器,USE_CUDA只是是否使用CUDA,由於我是要使用GPU,如果你不需要則不必加。

另外,下面的cmake命令則可以設置編譯選項,比如是否使用GLOG、GFLAGS等等第三方庫,由於我本人是要使用Caffe2來部署訓練好的模型,所以我只開啟了USE_OPENCV,如下圖所示。

接着以管理員身份打開vs2015開發人員命令提示,即Developer Command Prompt。使用cd命令至pytorch的script文件夾下,然后運行build_windows.bat,編譯需要稍長的時間。

編譯成功后,在pytorch文件夾下的build文件夾里,使用vs打開Caffe2.sln。我選擇了Release x64平台下,右鍵ALL_BUILD生成解決方案。然后再右鍵INSTALL生成解決方案,這兩個過程也是比較久的。INSTALL編譯成功后,在C:\Program Files就出現了Caffe2文件夾,里面主要存放了生成的include、lib和bin。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM