Window10 安裝 TensorFlow + CUDA8.0 + cudnn5.1

原創

SmileLing~

2018-08-25 04:04

安裝Anaconda
- 並配置安裝Python3.5版本，因爲TensorFlow在windows中只支持 Python3.5的版本
- 自己因爲之前安裝VS2017時候已經安裝好了Anaconda，這裏就不再安裝，沒安裝Anaconda的，可以參考下面的博客：Win10 TensorFlow（gpu）安裝詳解
安裝顯卡驅動：
- NVIDIA Driver Downloads
- 選擇默認精簡安裝即可. Installs all CUDA components and overwrites current Display Driver.
安裝CUDA：
- 下載：CUDA Toolkit 8.0 - Feb 2017
  - 選擇CUDA Toolkit 8.0 - Feb 2017，選擇自己系統win10，選擇本地安裝。接着我們下載以下兩個文件，基礎安裝包 Base Installer 1.3GB 和 補丁包 Patch 2 (Released Jun 26, 2017)。下載好後，只需要雙擊運行安裝第一個1.3GB的cuda_8.0.61_win10.exe ，都默認安裝就行。
- 驗證CUDA 8.0 是否安裝成功
  - 打開cmd 輸入：nvcc -V ，若如下所示，則說明安裝成功。
安裝cuDNN：
- Tutorial
- Download cuDNN v5.1 (Jan 20, 2017), for CUDA 8.0
將以上文件都下載好，解壓後，分別將三個文件放到cuda的相應安裝目錄下：

C:\cuda\bin\cudnn64_5.dll —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

並將 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\cudnn64_5.dll 配置到PATH環境變量中
安裝 TensorFlow
- 原生 pip 安裝
如果沒有安裝Python3.5 ，需要將其先安裝上。TensorFlow 在 Windows 上目前只支持Python 3.5.x。注意 Python 3.5.x 使用 pip3，我們用 pip3 來安裝 TensorFlow。
- 安裝只支持 CPU 的 TensorFlow，在 terminal 中輸入如下命令:
```
C:\> pip3 install --upgrade tensorflow
```
- 安裝支持 GPU 的 TensorFlow，使用如下命令：
```
C:\> pip3 install --upgrade tensorflow-gpu
```
- Anaconda 安裝 ( 簡單方便，自己推薦！)
（Anaconda 安裝是社區支持，而非官方支持）
- 按說明下載並安裝 Anaconda：
  - Win10 TensorFlow（gpu）安裝詳解或者 Anaconda download site
  - 配置Anaconda環境變量於PATH中：
- 建立一個命名爲tensorflowde 的conda 環境，指定Python版本是3.5 。通過打開系統命令行，Win+X+A ，然後再控制檯輸入：
```
conda create -n tensorflow python=3.5
```
- 查看當前環境
```
conda info -e
```
```
可以看到，我們當前所在的是默認的root環境。
```
- 激活 anaconda 環境：
```
activate tensorflow
(tensorflow)C:\>  # 你的提示符應該發生變化 
```
  可以看到，我們已經激活tensorflow環境
- 然後在tensorflow 這個 conda 環境中安裝只支持 CPU 的 TensorFlow（寫在一行）：
```
(tensorflow)C:\> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.1.0-cp35-cp35m-win_amd64.whl 
```
- 安裝支持 GPU 的 TensorFlow（寫在一行）：
```
(tensorflow)C:\> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.1.0-cp35-cp35m-win_amd64.whl 
```
補充知識 conda基本命令（具體細節可參考印象筆記）
- conda基本命令：
  - conda --version 查看當前Anaconda版本
  - conda info -e 查看當前環境版本
  - conda create -n python27 python=2.7 創建一個python2.7版本的開發環境
  - activate python27 切換到 python2.7的版本分支（環境）
  - deactivate 切換回 root分支（環境）
  - conda remove -n python27 --all 刪除一個分支（環境）
  - conda create -n clonedpython27 --clone python27 克隆python27這個分支，並將新分支命名爲clonedpython27
驗證安裝結果
- 安裝完TensorFlow以後，我們繼續在當前terminal中啓動python。
```
python
```
- 在 Python 交互式環境中輸入
```
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
```
- 如果系統輸出如下，則安裝成功：
```
Hello, TensorFlow!
```
在PyCharm中配置TensorFlow環境
- 要在pycharm下使用tensorflow，要設置好pycharm下解釋器interpreter的路徑File --> Setting --> Project --> Project Interpreter，這裏也就是tensorflow的路徑，如下圖所示：
- 在PyCharm中測試該代碼
```
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
```
運行後會提示警告，但這個警告並沒有影響

W c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\platform\cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use FMA instructions, but these are available on your machine and could speed up CPU computations.

我們可以通過在上面代碼之前，添加兩條語句，忽略掉警告
```
import os  ##加上這兩句，可以不顯示警告
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
```
Reference