之前整理了一篇在linux上安裝Tensorflow GPU開發環境的部落格,現在將Windows上配置的方式也整理一下

安裝準備

作業系統: Windows 10

Python版本: 3.6.5

TensorFlow-GPU版本: 1.6.0

Cuda版本: cuda_9.0.176_win10.exe

Cudnn版本: cudnn-9.0-windows10-x64-v7.zip

以上為我的安裝環境所選擇的安裝包的版本,

安裝時安裝包的版本對應關係可以見: https://tensorflow.google.cn/install/source_windows

安裝過程

1. 安裝python,一直下一步即可

2. 安裝Cuda,一直下一步,安裝完成之後可以看到,環境變數中出現:

在Path中配置為(已經有的不用再次配置):

點選確定

在dos控制檯輸入: nvcc -V,(輸入命令時注意V要大寫)出現:

說明Cuda安裝成功

3. 安裝Cudnn,解壓cudnn-9.0-windows10-x64-v7.zip

將Cudnn包中所帶的檔案放到Cuda目錄(上面的CUDA_PATH的路徑)下對應的資料夾即可

注意: 不要直接覆蓋目錄,是將檔案進行拷貝,根據資料夾

     Cudnn資料夾下的: bin/cudnn64_7.dll 拷貝到 Cuda資料夾下的 bin/ 目錄下

     Cudnn資料夾下的: include/cudnn.h 拷貝到Cuda資料夾下的 include/ 目錄下

     Cudnn資料夾下的: lib/x64/cudnn.lib 拷貝到Cuda資料夾下的 lib/x64/cudnn.lib目錄下

Cudnn解壓後的資料夾:

Cuda的目錄下的資料夾:

4. 安裝tensorflow-gpu

pip install tensorflow-gpu==1.6.0

安裝過程中可能會因為網路原因導致安裝中斷,多試幾次,安裝成功之後,開啟dos控制檯,進行測試

> python

> import tensorflow as tf

> a = tf.constant(1)

> b = tf.constant(2)

> sess = tf.Session()

> print(sess.run(a+b))

3

可以正常輸出結果,則tensorflow安裝成功

問題點

在安裝過程彙總有些機器在安裝Cuda時出現了

“Visual studio integration failed”的問題, 具體報錯如下:

解決方案,參照部落格: https://blog.csdn.net/roger17/article/details/81068030 

後順利安裝