win10 + GTX 1050 + Anaconda3.4.2.0 + tensorflow-gpu(0.12.0rcl)安装总结与问题解决

多一些分享,多一些美好!

经历过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.

2.可能遇到的问题

      2.1 安装cuda 的时候,在系统检查界面报错。

                                           win10 + GTX 1050 + Anaconda3.4.2.0 + tensorflow-gpu(0.12.0rcl)安装总结与问题解决_第1张图片

             我的解决方法是先忽略,但是在选项中一定选择自定义安装,并且只选择安装cuda一项。

                                              win10 + GTX 1050 + Anaconda3.4.2.0 + tensorflow-gpu(0.12.0rcl)安装总结与问题解决_第2张图片

       2.2  import TensorFlow as tf的时候提示:

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 "", line 978, in _gcd_import
File "", line 961, in _find_and_load
File "", line 950, in _find_and_load_unlocked
File "", 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啦!!!撒花 ~~~~

win10 + GTX 1050 + Anaconda3.4.2.0 + tensorflow-gpu(0.12.0rcl)安装总结与问题解决_第3张图片

        还有问题可以参考: http://blog.csdn.net/Kexiii/article/details/77990459


你可能感兴趣的:(tensorflow)