1. 程式人生 > >【轉】Win10 64 位Tensorflow-gpu安裝(VS2017+CUDA9.2+cuDNN7.1.4+python3.6.5)

【轉】Win10 64 位Tensorflow-gpu安裝(VS2017+CUDA9.2+cuDNN7.1.4+python3.6.5)

原文連結:https://blog.csdn.net/wwtor/article/details/80603296

Tensorflow 入門系列的第一篇,Tensorflow-gpu的下載與安裝。CSDN上能搜到很多學習資料,對初學者非常有幫助。剛安裝了目前最新版本的Tensorflow,當然也碰到不少坑,這篇文章詳細記錄了安裝過程,以供參考。

一 本機配置

  •  Win10 64bit (版本1803)
  • AMD Ryzen 5 2400G with Radeon Vega Graphics
  • NVIDIA GeForce GTX 1060

二 檔案下載

三 安裝過程

  • VS2017

選擇自己想要安裝的工作負載,也可以隨便選,有需要的時候再修改安裝。

詳細安裝步驟不再贅述,一路預設。

 

  • CUDA9.2

選擇相對應的版本下載,線上安裝版1.5GB,離線安裝版14.5MB,先安裝基礎包,再安裝升級補丁。

使用管理員許可權安裝,一路預設。

安裝完成之後開啟命令列,輸入:nvcc -V檢視版本,如果出現以下類似資訊表示安裝成功。

安裝成功後,我的電腦上點右鍵,開啟屬性->高階系統設定->環境變數,可以看到系統中多了CUDA_PATH和CUDA_PATH_V9_2兩個環境變數。

我們還需要新增如下幾個變數:

在使用者變數Path下新增:C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.2

在系統變數中新增如下幾個變數:

  •  CUDA_SDK_PATH = C:\ProgramData\NVIDIACorporation\CUDA Samples\v9.2
  •  CUDA_LIB_PATH = %CUDA_PATH%\lib\x64 
  • CUDA_BIN_PATH = %CUDA_PATH%\bin 
  • CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64 
  • CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64
  • 設定完成之後,我們可以開啟命令列來檢視。

     

  • cuDNN
  • 選擇對應的版本下載,解壓cudnn-9.2-windows10-x64-v7.1.zip,將資料夾裡的內容拷貝到CUDA的安裝目錄並覆蓋相應的資料夾,CUDA預設安裝目錄:

    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

  •  

  • Anaconda
  • 選擇相應的版本下載,這個下載過程有點慢,也可以選擇從清華映象下載,點選Anaconda3-5.2.0-Windows-x86_64.exe安裝,注意安裝過程中選擇“加入到系統PATH環境變數”。

     

  •  

  • Python
  • 搭建TensorFlow環境,官網下載Python3.6.5安裝,安裝的時候注意把“Add Python 3.6 to PATH”選項勾選上,這樣直接把使用者變數新增上了,後續不用再設定環境變數。

    此外,Anaconda集成了許多第三方庫,也可以從Anaconda自帶的Python3.6安裝。

    管理員模式開啟命令列,輸入:

    conda create –n tensorflow python=3.6    下載python並安裝。

     

    安裝完成後,輸入:conda list 可以查詢現在安裝了哪些庫,如果還有什麼包沒有安裝上,可以執行:conda install ***來進行安裝(***為需要安裝的包名稱)。如果某個包版本不是最新的,執行:conda update ***就可以更新了。

     

    啟用TensorFlow,輸入命令:activate tensorflow,檢視是否成功可以輸入:conda info –envs。

     

  • TensorFlow
  •  

    管理員模式開啟命令列,輸入命令:pip install tensorflow-gpu,如果選擇安裝cpu版本,命令:pipinstall tensorflow。注意,一定不要tensorflow-gpu和tensorflow(cpu版)一起裝,因為這樣裝有個先後順序問題,先安裝tensorflow-gpu再安裝tensorflow,gpu版本直接不能用了,你用withtf.device("/gpu":0),還是不好使,系統只會用cpu版本計算。而且這時候卸掉tensorflow還是不行,會報缺少檔案的錯誤,必須把tensorflow-gpu版的也卸掉,然後重新安裝tensorflow-gpu才行。解除安裝tensorflow,cmd命令列輸入:pip uninstall tensorflow,解除安裝tensorflow-gpu,輸入:pip uninstall tensorflow-gpu。

    目前tensorflow還不能支援cuda9.2,所以只能通過原始碼編譯或者別人編譯好的安裝包安裝。大神編譯的各種版本的tensorflow安裝包地址:

    https://github.com/fo40225/tensorflow-windows-wheel

    下載完成後將tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl

    複製到C:\Windows\System32資料夾下,然後執行命令:pip install tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl

     

  • 測試TensorFlow
  • https://blog.csdn.net/ygjustgo/article/details/78883981
  • https://blog.csdn.net/huangx06/article/details/78835360
  •  

    安裝完成後命令列輸入:python,進入python編輯環境,然後輸入命令:

    import tensorflow as tf

    安裝成功!

     

     

    參考資料