經歷過Ubuntu16.04+caffe的挫敗,我轉投了windows + tensorflow的懷抱,折騰了6個小時左右,成功配置了gpu加速環境,算是我個人計算機領域的一個小入門,在此記錄我踩過的坑坑~~~mark
不對之處,敬請指教!
以官方文檔作爲指南https://www.tensorflow.org/install/install_windows
xushiluo:http://www.jianshu.com/p/c245d46d43f0
mys異世界:http://www.jianshu.com/p/0c57e8e951af
兩篇文章對於安裝的過程說的已經很詳細了,就不再贅述,本文主要進行一個梳理,並給出可能遇到問題的解決方法。
0.安裝預備
安裝Rapid Environment Editor
安裝DXSDK_Jun10.exe
安裝Anaconda3.4.2.0
1. 安裝過程(特別注意版本對應,本小白就踩了坑坑~):
1.1 安裝vs2015 community版本
1.2 安裝cuda8.0,應該是8.0.44和8.0.61都是可以的,確保編譯成功之後進行後續操作。
1.3 安裝cuDNN-5.1,直接把安裝包解壓到CUDA\v8.0裏就行。
1.4 pip安裝tensorflow(python3.5的anaconda是不需要環境**的),這裏一定要注意不能直接安裝最新版,會出現找不到鏈接庫的問題,詳見下面的問題2.
我的解決方法是先忽略,但是在選項中一定選擇自定義安裝,並且只選擇安裝cuda一項。
Traceback (most recent call last):
File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in swig_import_helper
return importlib.import_module(mname)
File "C:\Anaconda3\lib\importlib\__init__.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "<frozen importlib._bootstrap>", line 978, in _gcd_import
File "<frozen importlib._bootstrap>", line 961, in _find_and_load
File "<frozen importlib._bootstrap>", line 950, in _find_and_load_unlocked
File "<frozen importlib._bootstrap>", line 648, in _load_unlocked
return _bootstrap._gcd_import(name[level:], package, level)
ModuleNotFoundError: No module named '_pywrap_tensorflow_internal'
Failed to load the native TensorFlow runtime.
See https://www.tensorflow.org/install/install_sources#common_installation_problems
for some common reasons and solutions. Include the entire stack trace
above this error message when asking for help.
這個的解決方法我是在評論裏找到的,感謝李廣飛,解決方法如下:
1、該鏈接提到的環境變量設置問題,確保你的path裏有C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin 和 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
2、就是tensorflow的版本問題,聯網的話直接pip install tensorflow-gpu會安裝最新版本,就會與樓主的tensorflow_gpu‑0.12.0rc1‑cp35‑cp35m‑win_amd64.whl不一致,最好時網上安裝用pip installhttps://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc1-cp35-cp35m-win_amd64.whl或者下載樓主的文件離線安裝。
此時,我的電腦已經可以成功跑起gpu的tensorflow啦!!!撒花 ~~~~
還有問題可以參考: http://blog.csdn.net/Kexiii/article/details/77990459