在win10下安裝最新Tensorflow1.6(CUDA9.0+cuDNN7.0)

本次安裝參考了tensorflow官網的安裝教程和JohnieLi的博客http://blog.csdn.net/johinieli/article/details/76960112。



以上是官網對於本次安裝提出的必備的硬軟件要求.

  • http://www.nvidia.com/object/cuda_gpus.html確認你的顯卡支持 CUDA。(若在列表中則支持CUDA編程)
  • 建議安裝 Anaconda,因爲這個集成了很多科學計算所必需的庫,能夠避免很多依賴問題

1.(安裝cuda之前建議先安裝vs2013或2015,不建議2017)首先安裝cuda,下載地址可從cuda官網下載。這裏下載的是cuda_9.0.176_win10.exe。下載好後雙擊默認安裝即可。

安裝完之後CUDA_PATH_V5_0和CUDA_PATH這兩個系統變量會自動爲你添加上。

環境變量配置: 
cuda8.0安裝完成之後在系統環境變量中自動配置了兩個系統變量 
CUDA_PATH:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 
CUDA_PATH_V9_0:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 
但是爲了之後的vs2013(或15)的配置做準備我們需要在配置五個系統變量 
CUDA_BIN_PATH:%CUDA_PATH%\bin 
CUDA_LIB_PATH:%CUDA_PATH%\lib\Win32 
CUDA_SDK_BIN:%CUDA_SDK_PATH%\bin\Win64 
CUDA_SDK_LIB:%CUDA_SDK_PATH%\common\lib\x64 
CUDA_SDK_PATH:C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0 
在系統環境變量path後添加如下內容 

;%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;

2.安裝cuDNN,將下載到的文件解壓放到任何一個目錄下就行,然後把你放的那個目錄添加到Path 環境變量裏。如C:\cuda。cuDNN同樣要在官網下載,下載前需填寫註冊問卷。我下載的是cudnn-9.0-windows10-x64-v7.zip,等上傳到網盤後可以在這裏提供一個下載鏈接。

3.安裝TensorFlow

# GPU版本
pip install --upgrade tensorflow-gpu

# CPU版本
pip install --upgrade tensorflow

等tensorflow安裝好了,整個過程就ok了。

注意:

到這裏如果你 import tensorflow 出現一堆報錯信息的話,可以嘗試這樣解決:將這些文件複製到相應的位置,

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

除此之外,還有一個辦法就是將C:\cuda\bin 也加進Path 環境變量裏,經過測試這樣也是可行的。

然後再次import tensorflow 應該就成功了。

感謝原博主提供的這個解決方案,我就是這樣成功的。


--

出現了Couldn't open CUDA library cupti64_90.dll的問題,這個問題是在使用tf.summary時出現的。在github中的一個issue裏找到了答案:



參考資料:

http://blog.csdn.net/johinieli/article/details/76960112

https://www.tensorflow.org/install/install_windows?hl=zh-cn



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章