Caffe windows 版本 安裝

本教程是caffe在windows系統上的配置方法,而且是github上官方BVLC/caffe目前推薦的配置方法,並不是使用微軟的caffe-windows

前期準備

  1. 首先從BVLC/caffe上下載caffe源碼,選擇好路徑後下載。下載過後,會得到一個caffe-windows.zip文件。
  2. 將該文件解壓到你指定的位置,進行下一步。
  3. 確定你是不是有NVIDIA的GPU,需不需要用GPU加速,如果不用GPU加速的話,可以跳過下面的cuda安裝和cudnn配置,直接配置caffe;如果需要用GPU,則需要在NVIDIA官網上分別下載安裝cuda和cdunn,下載地址在下面給出。
  4. 按照官方建議,需要使用cmake,所以需要先下載 cmake並安裝,給出的這個cmake鏈接是64位的,直接下載安裝,全英文界面,所有選項保持默認即可,一直點擊下一步,直到安裝完成。

環境配置

cuda和cudnn配置(不是使用NVIDIA GPU加速的話請掠過)

cuda和cudnn都是NVIDIA(英偉達)公司提供給開發者使用的運算平臺,簡而言之就是用GPU幹你想幹的事。

  1. 下載並安裝cuda,一定要選擇對應的版本,要不然會安裝失敗的。現在我演示的是64位win7,所以就選擇對應的版本下載。安裝過程比較簡單,雙擊安裝,選項建議都默認(有些人不喜歡把東西放在C盤,建議還是放在C盤,畢竟是開發工具),一直點擊確認或者下一步,直至安裝成功即可。 
  2. 下載cudnn,這是用來加速深度網絡訓練的。下載cudnn需要登陸,你可以直接用郵箱註冊一個,然後登陸下載就好了。
  3. 下載完後,你會發現,cudnn不是一個軟件,而是一個壓縮包,我下載的就是:cudnn-8.0-windows10-x64-v6.0.zip(有人私信說caffe還是沒有編譯成功,我看了一下發現是cudnn版本問題,前面的cuda安裝的是8.0,所以cudnn應該用的是cudnn-8.0-windows10-x64-v6.0.zip,而不是其他版本)。這個包需要解壓縮,得到一個名爲cuda的文件夾,爲了不與前面安裝的cuda混餚,用紅色表示。
  4. 配置cudnn。cudnn解壓後得到的cuda文件夾下有bin、include、lib三個子文件夾,現在需要將這三個文件夾下的東西分別放到之前cuda的安裝目錄下的對應文件夾下。之前已經安裝過cuda,cuda的安裝目錄默認爲:C:\Program Files\NVIDIA GPU Computing Toolkit\,現在就只需要: 
    -將cudnn解壓後得到的cuda文件夾下的bin下的cudnn64_6.dll(cudnn後面的數字可能不一樣,這就是版本問題,如果是cuda 8.0 建議搭配我這個版本的cudnn,否則可能會出現問題)文件拷貝到cuda安裝目錄下的bin目錄下; 
    -將cudnn解壓後得到的cuda文件夾下的lib文件夾下的x64文件夾下的cudnn.lib拷貝到cuda安裝目錄下的lib文件夾下的x64目錄下 
    -將cudnn解壓後得到的cuda文件夾下的include文件夾下的cudnn.h拷貝到cuda安裝目錄下include文件夾下 
    可能上面講的可能有點囉嗦,但是希望你們能看懂啦。
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章