windows安裝tensorflow GPU


一、安裝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交互窗口


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM