1. 程式人生 > >windows下caffe GPU版本配置

windows下caffe GPU版本配置

得到 詳細步驟 drive nbsp ubunt cudnn www uget caff

由於項目需要,所以在自己本子上配置了一下windows下GPU版本的caffe;

硬件: win10 ; gtx1070獨顯(計算能力6.1)

安裝軟件: cudnn-8.0-windows10-x64-v5.1 cuda_8.0.61_win10NugetPackages.zipcaffe-master

可以自己官網下載(我也提供了百度雲:鏈接:https://pan.baidu.com/s/1miDu1qo 密碼:w7ja)

參考鏈接: https://www.cnblogs.com/king-lps/p/6553378.html

(主要按照這篇和下面兩篇來配置的)

一個word版的配置教程:鏈接:https://pan.baidu.com/s/1hsEiDEG 密碼:d29u

http://www.cnblogs.com/zf-blog/p/6139044.html

http://blog.csdn.net/tina_ttl/article/details/51722983

https://www.2cto.com/kf/201612/575865.html

具體的詳細步驟我就先不寫了,之前一直在ubuntu上用的caffe,這次配置很輕松,我只說一下幾點註意的地方:

(1)安裝cuda時,註意 要取消勾選安裝driver compoent選項;

(2)cuda安裝好後會自動添加到系統變量中,而安裝完cuda後最好驗證一下(nvcc -V和運行deviceQuery.exe,兩者最好都跑一下,後者我運行時出現閃退情況,嘗試解決失敗,但也沒有影響最後的安裝);

(3)下載解壓完cudnn後,要將解壓後得到的三個文件夾合並到cuda相應的文件夾中就OK了;

(4)下載caffe時主要要下載windows版本的caffe,就是文件夾中要有windows這個文件夾;

(5)編譯的時候不需要全部編譯,編譯順序為libcaffe -> caffe ,如果需要再編譯python或者matlab接口;

然後就OK了,試試最簡單的mnist就行了;

windows下caffe GPU版本配置