1. 程式人生 > >win10安裝GPU版caffe記錄

win10安裝GPU版caffe記錄

微軟的caffe支援cuda8.0+cudnn v5.0的編譯

此處可以下載各種cuda版本:https://developer.nvidia.com/cuda-toolkit-archive
我選擇CUDA Toolkit 8.0 - Feb 2017版本
cudnn下載地址:https://developer.nvidia.com/cudnn
選擇 cuDNN v5 (May 27, 2016), for CUDA 8.0
1.安裝CUDA;
2.拷貝CPU版本的caffe和NugetPackages到C盤下;
3.修改配置檔案路徑C:\caffe\windows下的CommonSettings.props。(若擔心出問題,可將cudnn改為false)
在這裡插入圖片描述


4.將cudnn解壓到C盤:C:\cudnn-8.0-windows10-x64-v5.0-ga
5.新增 系統變數 CuDnnPath 變數值為C:\cudnn-8.0-windows10-x64-v5.0-ga
在這裡插入圖片描述
6.為了不出現什麼麻煩,將解壓的C:\cudnn-7.0-win-x64-v4.0-prod\cuda中的bin、include、lib三個資料夾拷貝到環境變數中CUDA_PATH顯示的路徑中的資料夾去,我的拷貝路徑是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
7.編譯caffe.sln。首先先重新編譯libcaffe。