一、安裝Anaconda
Anaconda是Python發行包,包含了很多Python科學計算庫。它是比直接安裝Python更好的選擇。
二、安裝Tensorflow
如果安裝了tensorflow,默認是CPU版的,要想安裝GPU版的,必須先pip uninstall tensorflow
卸載干凈之后,再使用pip install tensorflow-gpu
來安裝。
tensorflow CPU版比較簡單,安裝之后直接運行就可以。
tensorflow GPU版需要依賴CUDA,具體安裝見下文。
三、CUDA和cudnn的安裝
GPU是並行計算中必不可少的東西,在這一領域Nvdia處於領先地位。神經網絡這種計算量巨大、適合並行化,正是需要GPU的場景。
Tensorflow GPU版依賴CUDA,而cudnn是連接tensorflow和CUDA的紐帶。所以需要去Nvdia官網上下載CUDA和cudnn。CUDA安裝包比較大,大概1G多;cudnn安裝包下載前需要注冊用戶填寫調查問卷。需要注意的是,CUDA和cudnn的版本必須得匹配。
另外,Tensorflow用到的cudnn庫可能不是最新的,所以如果下載最新版的cudnn很有可能tensorflow無法識別。
下載完cudnn之后,解壓得到三個文件夾bin、include、lib,將其覆蓋復制粘貼到CUDA的安裝目錄下即可。
要想讓tensorflow看見cudnn.dll,就要讓CUDA的bin目錄添加到PATH環境變量中去。
CUDA安裝完成后,需要重啟電腦。
可能遇到的異常
Cannot remove entries from nonexistent file c:\anaconda\lib\site-packages\easy-install.pth
解決方案
pip install --upgrade --ignore-installed setuptools
Blas SGEMM launch failed
解決方案:關閉其他Python交互窗口