1. 程式人生 > >windows下tensorflow-GPU版本安裝

windows下tensorflow-GPU版本安裝

隨著tensorflow的更新的,windows下的GPU版本也可以愉快的安裝了,以前只能安裝cpu。其實網上這方面的教程很多,但是搜尋出來的基本都是16年的部落格,安裝步驟繁瑣。在現在看來根本不需要那麼麻煩,下面我就介紹一下大致流程,以前自己用的比較nice的網址。

第一步:下載anaconda

這個不用說了吧,很簡單直接百度anaconda,下載官網釋出的最新版本的anaconda,根據教程安裝好,其實很簡單,基本就是雙擊一路點下去的事,只是其中有一步,需要把兩個小方框的對號都點上,這是為了環境變數的配置。其他的都是預設,so easy。安裝好之後,cmd,進去windows的終端。輸入python命令,看看是否能進入python,確保你的python環境已經安裝好。


第二步,安裝cuda,cudnn

這裡的教程我就不說了,網上多的不行,windows下的cuda和cudnn安裝基本都是雙擊的事情

第三步,安裝tensorflow的GPU版本或者CPU版本

這裡很簡單,沒有網上說的那麼麻煩,下載這個,下載那個。直接輸入:

pip install tensorflow-gpu
就可以安裝了,當然這裡如果你要裝的是cpu版本的,直接除去命令後面的-gpu就行了 。但是,最可能出現的問題是,下載包的速度太慢,怎麼辦?換下載源啊,換下載源這事,網上教程多的是,這裡我貼一個部落格的網址,大家自己可以去看看,也是簡單的不要不要的:點選開啟連結 那麼我們接下來換個命令安裝tensorflow:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu
下面就是安心等待吧,很快tensorflow的gpu版本就會安裝好。

至此本部落格的主要內容就結束了,但是額外提供一點福利,我們裝好python,裝好tensorflow之後,在跑某些github上的程式碼或者其他的深度學習程式碼時,經常出現缺少這個模組那個模組問題。怎麼辦?首先用上面的pip 加-i https://pypi.tuna.tsinghua.edu.cn/simple 的方式安裝需要模組,若是還不行,出現下載失敗等問題,這裡提供一個網址python的模組網址,只要進去,下載相應的模組。然後在windows終端進入模組包的存放地址,利用 :

pip install 模組包的名稱
就可以安裝了,基本能解決大部分問題。python模組包的地址:
https://www.lfd.uci.edu/~gohlke/pythonlibs/


另外提供一下tensorflow的官方文件地址:http://devdocs.io/tensorflow~python/,有興趣的同學可以瞭解一下。