【轉】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 官網下載地址
- cuDNN7.1 官網下載地址(需要註冊)
- Anacoda5.2 官網下載地址
- Python3.6.5 官網下載地址
三 安裝過程
- 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
安裝成功!
參考資料