系統:Windows10
顯示卡:gtx 1050Ti
開發環境:vs2013
1、CUDA安裝
下載下傳位址:https://developer.nvidia.com/cuda-downloads
2、cudnn下載下傳
位址:https://developer.nvidia.com/rdp/cudnn-download
下載下傳後解壓會有三個檔案夾bin、include、lib。将這三個檔案夾複制到cuda的安裝目錄中:\NVIDIA GPU ComputingToolkit\CUDA\v8.0。(cuda的安裝目錄中也有這三個檔案夾,将這三個檔案夾分别與原來存在的檔案夾合并,如下圖所示)。
3、caffe-windows安裝
(1)下載下傳官方caffe-windows并解壓,将 .\windows\CommonSettings.props.example備份,并改名為CommonSettings.props。如下圖所示:
我的CommonSettings.props檔案更改的位置:
<CpuOnlyBuild>false</CpuOnlyBuild>
<UseCuDNN>true</UseCuDNN>
<CudaVersion>8.0</CudaVersion>
編譯caffe-windows
編譯用vs2013打開.\windows\Caffe.sln 并将解決方案的配置改為release,點菜單欄上的“生成->生成解決方案”,會将整個項目全部生成,這個時間會比較長(由于官方caffe-windows 的版本使用了NuGet管理第三方開發包,是以需要在vs2013上安裝NuGet,)。生成成功之後的檔案都在.\Build\x64\Release中。
到此安裝成功,繼續進行訓練可以參考我其他的部落格。