1. 程式人生 > >環境配置6-Win10下安裝CUDA和cuDNN,配置caffe

環境配置6-Win10下安裝CUDA和cuDNN,配置caffe

到官網下載相應的安裝包,cuDNN為lib庫,需要先註冊NVIDIA賬號(或者連結複製到迅雷下載)。

https://developer.nvidia.com/rdp/cudnn-archive

https://developer.nvidia.com/cuda-80-ga2-download-archive



1-cuda下載好後,直接按照預設選項安裝。出現下面的提示直接忽略。


2-cudnn下載後解壓(比如解壓後為C:\cuda\),會有bin、include、lib三個資料夾,將這三個資料夾下的內容複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下對應的資料夾中。

(以下3、4、5步驟是否還需要做待驗證,可以先不做,安裝caffe後再觀察)

3-新增cudnn環境變數。

    將C:\cuda\bin新增到系統環境變數。

    右擊我的電腦——屬性——高階——環境變數——系統變數——PATH

4-開啟visual studio專案

    選擇release、x64

    除錯——屬性


VC++Directories——Include Directories中新增C:\cuda\include

VC++Directories——Library Directories中新增C:\cuda\lib\x64


Linker——Input——Additional Dependencies中新增cudnn.lib


接下來可以安裝caffe

1-下載Windows下的caffe

https://github.com/BVLC/caffe/tree/windows

2-解壓出來,用notepad++開啟

caffe——scripts——build_win.cmd

3-if DEFINED APPVEYOR不用管,直接看else後的。

MSVC_VERSION=14  (vs2015)

WITH_NINJA=0

CPU_ONLY=0  (如果用GPU)

PYTHON_VERSION=2  (根據安裝的python版本)

其餘的不用改變,儲存後退出。

4-雙擊執行build_win.cmd開始編譯,等待十幾分鍾。


5-編譯成功。將生成的build——lib——release資料夾下的10個檔案,拷貝到程式裡win_lib下替換。