1. 程式人生 > >win10+vs2013+caffe+gpu+python環境配置

win10+vs2013+caffe+gpu+python環境配置

前言

之前寫過win7+vs2013+caffe+cpu+python環境配置,但是這種低配置的電腦根本跑不動深度學習,所以最近買了一臺高配置電腦,今天研究了一天終於配置成功了,所以迫不及待的想過程和心得記錄下來。我的電腦配置如下所示。 作業系統:win10專業版 處理器: i5-8300H 記憶體:8G 顯示卡:GTX1600 6G獨顯

前期準備

  1. 安裝vs2013 網上也有教程說可以使用vs2015,但是我試過了,沒能安裝成功,可能是我技術不到家,我建議還是安裝vs2013最好。但是在安裝vs2013時因為win10版本高的原因,往往會出現部分功能未能正確安裝,不要以為這沒有影響,這會導致你安裝不上caffe,所以在安裝的時候不要選擇所有功能,只要選擇如下所示三個功能就行。 在這裡插入圖片描述
  2. 安裝cudn 在安裝cudn之前,首先得確定你的顯示卡驅動安裝正確,而且要下對應的版本,開啟NVIDIA控制面板,點選幫助,在點選系統資訊,點選元件,我的版本是cuda9.2,網上很多教程都沒有說到這一塊,所以我在這研究了一天才解決,你得根據你電腦的顯示卡下載對應的版本。下載好之後直接點選安裝預設路徑就行,接著在下載cuddn v4在這裡插入圖片描述 在這裡插入圖片描述
  3. 在安裝cuda時往往會出錯,主要是由於你電腦顯示卡驅動安裝的不對,最好去你電腦的官網下載最新的顯示卡驅動。cuddn下載後解壓得到如下三個資料夾, 在這裡插入圖片描述 三個目錄的檔案分別為 在這裡插入圖片描述 在這裡插入圖片描述 在這裡插入圖片描述 將相應的bin、include、lib分別放於自己的cuda下面的相應目錄中,cuda預設目錄為C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v7.5,因此將剛才解壓的檔案放在這個目錄下面的bin、include、lib資料夾下。例如將cudnn64_5.dll複製到bin目錄之後: 在這裡插入圖片描述
  4. 安裝Anaconda2 這網上安裝教程有很多就不在詳細敘述了。

caffe配置

當前面的工作都做完的時候,接下來可以開始配置caffe了

  1. 開啟資料夾D:\caffe\caffe-master\windows,找到CommonSettings.props.example檔案並將它複製並重命名為CommonSettings.props,如圖所示 在這裡插入圖片描述
  2. 用vs2013開啟CommonSettings.props檔案,將第9行原來的7.5改為你cuda的版本,我的版本是9.2,在前面說過。 在這裡插入圖片描述 將第13行該為true 在這裡插入圖片描述 將第48行改為Anaconda2的安裝路徑 在這裡插入圖片描述 儲存關閉檔案
  3. 開啟D:\caffe\caffe-master\windows資料夾下Caffe.sln 在這裡插入圖片描述
    將活動配置改為Release模式 在這裡插入圖片描述 右擊解決方案,點選成成解決方案 在這裡插入圖片描述
  4. 編譯成功後開啟D:\caffe\caffe-master\Build\x64\Release檔案如果出現pycaffe資料夾,以及caffe.exe檔案,則說明你已經編譯成功 在這裡插入圖片描述 開啟pycaffe資料夾,複製caffe資料夾,如圖所示。 在這裡插入圖片描述 將caffe資料夾複製到Anaconda2安裝目錄下D:\mysoft\Anaconda2\Lib\site-packages,這是我的安裝路徑,你得找到你電腦的安裝路徑,如圖所示。 在這裡插入圖片描述
  5. 開啟cmd命令視窗,輸入python,(由於我的電腦只安裝了Anaconda2,只有python2.7版本,如果你電腦有多個python版本的話,需要找到你指定的目錄,這一步過程不在敘述)輸入import caffe,如圖所示。 在這裡插入圖片描述 這時會出現 ImportError: No module named google.protobuf.internal 這個錯誤。原因是沒有protobuf這個模組。使用pip install protobuf(這個安裝也需要注意,在cmd裡指定下目錄,調D:\mysoft\Anaconda2\Scripts\裡的pip.exe 即pip.exe install protobuf).如圖所示 在這裡插入圖片描述 模組安裝完成之後,再import caffe就ok了。如果沒有報錯的話,到此環境已經配置成功了。 如果你想測試你電腦效能的話,你可以用網上voc進行測試,我的這篇部落格https://blog.csdn.net/weixin_42795611/article/details/83479011有詳細介紹如何訓練資料集,感興趣的可以試一試。

總結

如果你是第一次安裝caffe的話,不可能一帆風順,總會出現各種問題,出現問題不可怕,只要一一去解決就行。我的郵箱是[email protected],如果你有什麼問題或者交流的話,可以隨時聯絡我。下一篇部落格將介紹如何製作自己的資料集。