1、下載並安裝Acaconda3(將Anaconda路徑添加至環境變量)
https://www.anaconda.com/download/
2、修改python版本
conda install python=3.6 //換成所需要的版本
3、創建虛擬環境
虛擬環境名稱自定義,我將虛擬環境名稱設置爲py27,python版本需要指定
conda create –n py27 python=2.7
創建的虛擬環境爲一個文件夾,在Anaconda安裝路徑下名爲envs的文件夾中。
4、激活虛擬環境
activate py27(自己創建的python虛擬環境名稱)
5、在虛擬環境下安裝opencv-contrib-python
https://pypi.org/project/opencv-contrib-python/ (Release history選擇需要的版本py)
caffe的安裝
6、下載文件,caffe源碼+必要第三方庫+cuda+cudnn
鏈接: https://pan.baidu.com/s/1WAColGir0z6WwHnu-sGa1w
提取碼: uivk
CPU版本只需要下載兩個文件:
GPU版本需要下載四個文件:
7、安裝cuda庫(GPU版本,如果只安裝CPU版本的可以跳過該步驟)
安裝cuda前最後先安裝VS2015,否則會出現“未找到導入的項目“C:\Program Files (x86)\MSBuild\Microsoft.Cpp\v4.0\V140\BuildCustomizations\CUDA 8.0. props”以及“libcaffe加載失敗”)
解決方法:將“C:\Program Files (x86)\MSBuild\Microsoft.Cpp\v4.0\V140\BuildCustomizations”目錄下文件放入下述文件文件夾。
A、雙擊“cuda_9.0.176_win10.exe”進行安裝,只勾選cuda模塊,其它可以不勾選(若電腦顯卡驅動沒有安裝是最好全選)。注意:最好關閉殺毒軟件,否則可能造成安裝失敗。
安裝完成後,解壓剛纔下載的“cudnn-9.0-windows10-x64-v7.1.zip”,得到“cuda”文件夾,將文件夾裏面的bin、include、lib三個文件夾複製到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0”目錄下,覆蓋掉,完成cuda與cudnn的安裝。
測試:打開cmd命令,輸入 nvcc -V 查看是否出現如下信息,有如下信息則表明安裝成功。
8、編譯Caffe源碼
1)解壓下載的“caffe-windows-ms.zip”文件夾和“thirdparty.zip”文件夾;
2)將“thirdparty”文件夾下面的所有子文件夾複製到“caffe-windows-ms\windows\thirdparty”文件夾下面;
3)進入“caffe-windows-ms\windows”文件夾下面,複製一份當前目錄下的“CommonSettings.props.example”文件,並重命名該文件爲“CommonSettings.props”;
4)用記事本或者notepad++等編輯軟件打開該文件,修改幾個地方:
GPU版本的修改地方與修改內容如下:
cuDnnPath設置爲解壓得到的目錄,如下爲解壓目錄:
CPU版本的修改地方與修改內容如下:
5)用visual studio2015打開當前目錄下的“Caffe.sln”,右擊解決方案“caffe”,如下設置:
6)配置Python環境(如果選擇支持python的話)
7)按照如下屬性依次生成項目(選擇release和x64),詳細步驟如下:
A、編譯libcaffe;
B、編譯caffe;
C、複製“caffe-windows-ms\windows\thirdparty\bins”文件夾下所有文件到“caffe-windows-ms\Build\x64\Release”文件夾下;
D、編譯pycaffe;
E、編譯convert_imageset(可選,用於將圖片轉爲mdb的工具);
F、編譯成功後,“caffe-windows-ms\Build\x64\Release”文件夾下面就會生成編譯好的文件;此時在當前目錄下打開cmd,輸入“caffe”,出現如下信息就說明成功了。(由於我是在anaconda虛擬環境下裝的python2.7,因此需要先輸入activate py27(py27爲創建的虛擬python環境名稱) 激活所創建的虛擬python環境)
G、將“caffe-windows-ms\Build\x64\Release\pycaffe”文件夾下面所有文件及子文件複製到所使用的python安裝目錄的“Lib\site-packages”(我的python安裝目錄爲 D:\software\anaconda\envs\py27\Lib\site-packages)下面就可以,打開python,輸入import caffe,查看是否成功導入。如果提示No module named ***,直接通過pip install ***即可。
安裝過程中可能遇到的問題,安裝的cuda10,所以需要自己重新編譯nccl.dll和nccl.lib:
如果提示“由於找不到cudart64_**dll,無法繼續執行代碼。重新安裝程序可能會解決此問題”。
解決方法:(1)下載nccl工程自己編譯下,地址https://github.com/happynear/nccl
(2)打開裏面的windows文件夾,修改工程屬性文件nccl.vcxproj,找到裏面兩個cuda版本號設置的地方,改爲10.0(自己的cuda環境)
(3)重新編譯生成nccl.dll和nccl.lib ,替換caffe文件夾中thirdparty裏NCCL\bins的的nccl.dll以及NCCL\bin的nccl.dll和NCCL\lib的nccl.lib。
如果載入過程提示錯誤:未找到導入的項目“C:\Program Files (x86)\MSBuild\Microsoft.Cpp\v4.0\V140\BuildCustomizations\CUDA 10.0.props”。
1.修改nccl-master\windows\nccl.vcxproj文件中的cuda版本號(第一個問題問題中以及修改)
2.、如果對應路徑下沒有該文件,則返回兩級文件夾,進入進入V120\BuildCustomizations可以找到對應的CUDA 8.0.props文件,然後拷貝到V140\BuildCustomizations即可成功。