本次安裝參考了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 出現一堆報錯信息的話,可以嘗試這樣解決:將這些文件複製到相應的位置,
- C:\cuda\bin\cudnn64_7.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:\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