標籤:

Windows10安裝tensorflow-gpu心得

閑來無事,賦閑在家,於是想在新買的遊戲本上部署一下深度學習環境,前前後後花了將近一個上午+下午的時間。

首先來介紹一下鄙人的電腦配置:

CPU i7-7700Hz

內存8G

GPU Geforce GTX 1060

安裝過程主要分為以下五個步驟:

1.安裝Python3.5對應的Anaconda 4.2.0.從這個網址里找:

Anaconda installer archive

註:這一步中建議勾選將Anaconda路徑加入環境變數操作。

安裝成功後,在DOS命令行中輸入 conda list 指令,如安裝成功,則顯示一下信息:

2.安裝CUDA8

CUDA Toolkit 8.0 - Feb 2017

安裝成功之後,在DOS命令行中輸入 nvcc -V 指令,如果安裝成功,則顯示以下信息:

註:這一步中建議使用默認安裝地址,不要自定義更改。

3.安裝cuDNN5.1

cuDNN Archive

這一步需要申請賬號,填寫一下資料,分分鐘的事。

下載一個壓縮包cudnn-8.0-windows10-x64-v5.1.zip,解壓之後得到一個cuda文件夾,我把這個文件夾拷貝在D盤下:

cuda文件夾下有三個文件夾:bin、include、lib;將這三個文件複製到

C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0 目錄下。

註:這一步還有一個操作,就是將解壓後cuda文件夾下的bin目錄加入到環境變數中:

4.安裝tensorflow-gpu:

在DOS命令行中:

使用python -m pip install --upgrade pip 指令將pip升級到最新版本;

使用pip install setuptools --ignore-installed 忽略已經安裝的setuptools版本,直接安裝最新的版本;

使用指令pip install tensorflow-gpu==1.2.0 安裝tensorflow-gpu版本。

註:第二個指令不要用 pip install --upgrade tensorflow-gpu 指令安裝tensorflow,因為這個指令安裝的是最新版本的tensorflow,與之前安裝的CUDA8cuDNN5.1存在不兼容問題。

如果想進一步安裝keras,同樣的使用pip install keras 指令即可。

在python編輯器中,使用 import tensorflow as tf [ 以及 import keras] 進行驗證是否安裝成功。

5.安裝Pychram

強烈推薦python的兩個IDE:Pychram以及Jupyter notebook,這倆利器真的是誰用誰知道,建議在編程開發中,二者結合使用。

後者在Anaconda3中能找到:

Pycharm在官網中下載:

Download PyCharm: Python IDE for Professional Developers by JetBrains

選中右側的社區版下載,有人會問為嘛不用左邊的專業版呢?

答案很簡單,專業版的收費啊,只有30天免費使用的時間。

一頓安裝結束之後,在Pycharm裡面小試牛刀一把:

使用tensorflow完成的加法運算,輸出30...成功了喲...

筆者註:

這裡著重強調各個組件的版本兼容性問題:

首先,windows下的tensorflow目前只支持python3.5以上的版本,而python3.5是python3中最穩定的版本,所以本文安裝Python3.5對應的Anaconda 4.2.0.。

其次,Anaconda 自帶的setuptools 以及pip版本較低,需要使用指令升級一下才能安裝tensorflow。

最後,tensorflow-gpu1.3.0以下的版本對應CUDA8,而CUDA8對應cuDNN5.1。

此外,tensorflow因為版本更新過於頻繁而飽受非議,且最新版本的tensorflow-gpu1.3.0網友都過有bug,所以建議直接安裝1.2.0版本。


推薦閱讀:

劉光聰 | TensorFlow:揭示多語言編程的奧秘
在 Keras 和 slim 中進行 debug
用Tensorflow自動化構建海洋生物系統,利用上萬的圖片訓練,找到瀕臨物種「海牛」是什麼原理?
TF里幾種loss和注意事項

TAG:TensorFlow |