经历过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 "
File "
File "
File "
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