1. 程式人生 > >Windows上安裝TensorFlow無坑教程

Windows上安裝TensorFlow無坑教程

TensorFlow是谷歌基於DistBelief進行研發的第二代人工智慧學習系統可被用於語音影象識別等多項機器深度學習領域。對於這種高大上的東西估計很多人都想學吧,包括我這位零基礎的小編。但網上找到的教程大多都不是新手向的,而且基本用的也不是Windows系統,這讓我們這種新手很苦惱。但我參考了Andrew Che的教程,加上實踐操作,我成功的在Windows上安裝了TensorFlow,以下是我的步驟:

1.檢查GPU是否支援CUDN​:其實這一步是可以省略的,因為CUDN支援的顯示卡範圍特別廣,你基本只要確認你的GPU是不是NVIDIA的就可以了,當然如果你還是不確定的話,也可以去這個網站(https://developer.nvidia.com/cuda-gpus)中的“CUDA-Enabled GeForce Products”查一下,只要你的顯示卡型號在裡面就可以了。

​2.下載並按照CUDA:進入此網站(https://developer.nvidia.com/cuda-downloads),點選Windows

選擇你的電腦的系統版本(這裡我用的是win7)

點選第一個選項

​選擇第一個下載

下載完之後開啟安裝,一路確認到底

​安裝完後進入下一步

3.下載cuDNN:​其實,到了這一步比較鬧心,因為當你進了下載網站(https://developer.nvidia.com/rdp/form/cudnn-download-survey),你會發現他要讓你註冊了才能下載,但為了能安裝,註冊什麼的就幹了吧,不過程比較煩

然後你就可以現在下載了。考慮到相容性,這裡我們就選擇第三個進行下載安裝

​下載後將其解壓縮

​在C盤根目錄下新建個tools資料夾,將解壓後的“cuda”資料夾放入其中

​4.設定環境變數:右鍵“我的電腦”,選擇“高階系統設定”

​進入“環境變數”

進入“Path”,新增"c:\tools\cuda\bin"與”C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin”後點擊確認

5.下載並安裝Anaconda:進入下載網站(https://www.continuum.io/downloads),點選Windows圖示

​這裡我們選擇版本為3.6的進行下載並安裝

下載後開啟進行安裝,安裝時這裡選擇第一個,也就是它推薦的選項

​然後將第兩個選項都點上

​安裝完後進入下一步

​6.設定TensorFlow工作環境:在你喜歡的地方新建個tensorflow目錄。這裡我就在c盤更目錄建立了。

​按win+r開啟“執行”輸入cmd,開啟命令提示符

輸入cd +​​tensorflow所在目錄,回車

​接下來依次輸入一下幾個命令,注意結尾不能有空格,如果途中讓你選擇是Yes/no,全部輸入y

conda create -name tensorflow-gpu python=3.5 anaconda

activate tensorflow-gpu​

pip install tensorflow-gpu

pip install keras

轉載注: 此處的幾個命令有坑,更具體的步驟參考我的一篇部落格:http://mp.blog.csdn.net/postedit/79358543

7.確認TensorFlow安裝完成:在命令提示符中輸入“jupyter notebook”,在瀏覽器中召喚jupyter notebook

新建​​個python3檔案

​在裡面輸入以下指令

import tensorflow as tf

tf.__version__

​按下shift+enter執行後假如看到1.2.0,說明你成功了,接下來你就可以自由用tensorflow了