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。