本機(jī)環(huán)境:Win10,Anaconda3,VS2015,GTX1080
Step 1:pip3 install tensorflow-gpu
Step 2:安裝CUDA_8.0.61和CUDNN_v6.1,這里要注意兩個(gè)版本要匹配,比如CUDA_8.0.61配CUDNN_v5.1就不行
Step 3:將解壓后的CUDNN中的文件按相應(yīng)的目錄復(fù)制到CUDA安裝的目錄中:
X:\cuda\bin\cudnn64_6.dll —> Y:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin?
X:\cuda\include\cudnn.h —> Y:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include?
X:\cuda\lib\x64\cudnn.lib —> Y:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
Step 4.設(shè)置環(huán)境變量,在Path中加上CUDNN解壓后的文件所在路徑
結(jié)尾?
嘗試在手寫(xiě)數(shù)字集內(nèi)測(cè)試仆百,迭代次數(shù)2w,GPU版本耗時(shí)3min,在tensorflow-CPU版本下花費(fèi)半個(gè)多小時(shí)硝烂。