Windows下Tensorflow-GPU的安裝


Windows下Tensorflow-GPU的安裝

一直在搞3D視覺相關的技術工作,最近需要用到機器學習(深度學習)來實現一些功能。而Tensorflow是谷歌發布的一款非常流行的機器學習庫,故嘗試在Pycharm下安裝Tensorflow的GPU版本。

安裝前准備

由於開發的環境是Windows與Pycharm,且使用Anaconda來對Python包進行管理,所以需要事先安裝好Anaconda與Pycharm IDE。同時配置好Anaconda與pip的鏡像源,推薦使用清華大學或豆瓣的源。上述軟件的安裝與配置教程請自行上網查閱,此處不作展開。

安裝cuda

因為要安裝Tensorflow的GPU版本,所以需要用到Nvidia的cuda工具。cuda的安裝包可以上nvidia官網上下載,不過不知為何,其下載速度非常慢且經常會掛掉,所以此處推薦使用百度雲的離線下載功能,如圖1所示,將官網上的下載鏈接復制到圖2中,然后保存到百度雲盤中,然后用百度雲盤下載即可,這樣下載就比較穩定且速度較快。下載完后直接安裝即可。譬如我的安裝路徑是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin,將該路徑添加到系統變量Path中。

 

圖1 百度雲中的離線下載

圖1 百度雲中的離線下載

 

 

圖2

圖2

 

安裝cudnn

cudnn是Nvidia開發的針對深度神經網絡加速的工具,其安裝方法比cuda簡單,從Nvidia下載,也可以從如下百度雲地址下載:
鏈接:https://pan.baidu.com/s/1c1GI1mo 密碼:oujf
然后直接解壓,里面的cuda文件夾中有三個文件夾,分別為:bin、include、lib,將這三個文件夾分別復制到cuda的安裝目錄中,在這里,應該將這三個文件夾復制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0。即可完成對cudnn的安裝。

安裝Tensorflow-gpu

以前在Anaconda中安裝tensorflow那是相當的簡單,直接在Anaconda Cloud中搜索tensorflow-gpu,即找到tensorflow-gpu的安裝腳本命令,在PyCharm的終端中輸入該命令,即可進行自動下載,只是耗時較長而已。但不知為何,現在這方法時行時不行,即下載相當不穩定,經常下到一半就出錯,所以現在改為用pip來下載。
其方法如下:打開PyCharm中的終端,輸入命令:

pip install tensorflow-gpu

然后輸入y,即可進行 tensorflow-gpu版本的下載,現在能下載的版本是1.4.0。

驗證

下載完成Tensorflow后,在PyCharm中Python Console中進行驗證是否安裝成功。輸入如下python代碼:

import tensorflow as tf
tf.__version__

輸出結果如下:

 

enter description here

圖3 tensorflow驗證結果

 

即證明Tensorflow-gpu安裝成功。


免責聲明!

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



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