Windows10 安装tensorflow GPU版

转载于：https://blog.csdn.net/listener51/article/details/77752027

windows10安装tensorflow的gpu版本（pip3安装方式）

前言：

　　　TensorFlow 有cpu和 gpu两个版本：gpu版本需要英伟达CUDA 和 cuDNN 的支持，cpu版本不需要；本文主要安装gpu版本。

 

1、环境 

• gpu：确认你的显卡支持 CUDA，这里确认 。
• vs2015运行时库：下载64位的，这里下载，下载后安装。
• python 3.6/3.5：下载64位的，这里下载，下载后安装。
• pip 9.0.1（确认pip版本 >= 8.1，用pip -V 查看当前 pip 版本，用python -m pip install -U pip升级pip 。）

注意：pip其实在安装python的时候就自带了，但是需要将其（XXX:\ Python\Python36\Scripts）配置到环境变量(path)中，红色字体代表python安装 的路径。

2、安装

（1）、下载 CUDA 
　　　　Google 官方推荐CUDA8 ，可以去官网下载：cuda

• 安装CUDA 

　　　直接双击exe安装，按提示下一步装到底就行。装完测试一下是否安装成功，命令行输入 nvcc -V ，看到版本信息就表示安装成功了。 

（２）、下载cuDNN

      　tensorflow版本不同，需要的cuDNN版本也不一样，具体看tensorflow的发布说明，如:tensorflow1.3的发布说明

• 配置cuDNN

　　　下载到对应版本的cuDNN（tensorflow1.3需要cuDNN6，可到 https://www.zhihu.com/question/37082272下载）后，解压：

1.   将解压后的bin目录配置到系统环境变量(path)中；
2.   将相应的文件拷贝到相应的目录：　　

        　　copy cudnn\bin\cudnn64_5.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\
　　　　copy cudnn\include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include\
　　　　copy cudnn\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64\

（３）、安装tensorflow

1. # GPU版本

2. pip3 install --upgrade tensorflow-gpu

3.  

4. # CPU版本

5. pip3 install --upgrade tensorflow

（４）、卸载tensorflow

pip3 uninstall tensorflow

3、测试

1. import tensorflow as tf

2.  

3. a = tf.random_normal((100, 100))

4. b = tf.random_normal((100, 500))

5. c = tf.matmul(a, b)

6. sess = tf.InteractiveSession()

7. sess.run(c)

 

issue:

1. Traceback (most recent call last):

2. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in swig_import_helper

3. return importlib.import_module(mname)

4. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in import_module

5. return _bootstrap._gcd_import(name[level:], package, level)

6. File "<frozen importlib._bootstrap>", line 986, in _gcd_import

7. File "<frozen importlib._bootstrap>", line 969, in _find_and_load

8. File "<frozen importlib._bootstrap>", line 958, in _find_and_load_unlocked

9. File "<frozen importlib._bootstrap>", line 666, in _load_unlocked

10. File "<frozen importlib._bootstrap>", line 577, in module_from_spec

11. File "<frozen importlib._bootstrap_external>", line 906, in create_module

12. File "<frozen importlib._bootstrap>", line 222, in _call_with_frames_removed

13. ImportError: DLL load failed: The specified module could not be found.

14.  

15. During handling of the above exception, another exception occurred:

16.  

17. Traceback (most recent call last):

18. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 41, in <module>

19. from tensorflow.python.pywrap_tensorflow_internal import *

20. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 21, in <module>

21. _pywrap_tensorflow_internal = swig_import_helper()

22. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 20, in swig_import_helper

23. return importlib.import_module('_pywrap_tensorflow_internal')

24. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in import_module

25. return _bootstrap._gcd_import(name[level:], package, level)

26. ImportError: No module named '_pywrap_tensorflow_internal'

27.  

28. During handling of the above exception, another exception occurred:

29.  

30. Traceback (most recent call last):

31. File "<pyshell#0>", line 1, in <module>

32. import tensorflow as tf

33. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\__init__.py", line 24, in <module>

34. from tensorflow.python import *

35. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\__init__.py", line 51, in <module>

36. from tensorflow.python import pywrap_tensorflow

37. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 52, in <module>

38. raise ImportError(msg)

39. ImportError: Traceback (most recent call last):

40. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in swig_import_helper

41. return importlib.import_module(mname)

42. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in import_module

43. return _bootstrap._gcd_import(name[level:], package, level)

44. File "<frozen importlib._bootstrap>", line 986, in _gcd_import

45. File "<frozen importlib._bootstrap>", line 969, in _find_and_load

46. File "<frozen importlib._bootstrap>", line 958, in _find_and_load_unlocked

47. File "<frozen importlib._bootstrap>", line 666, in _load_unlocked

48. File "<frozen importlib._bootstrap>", line 577, in module_from_spec

49. File "<frozen importlib._bootstrap_external>", line 906, in create_module

50. File "<frozen importlib._bootstrap>", line 222, in _call_with_frames_removed

51. ImportError: DLL load failed: The specified module could not be found.

52.  

53. During handling of the above exception, another exception occurred:

54.  

55. Traceback (most recent call last):

56. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 41, in <module>

57. from tensorflow.python.pywrap_tensorflow_internal import *

58. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 21, in <module>

59. _pywrap_tensorflow_internal = swig_import_helper()

60. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 20, in swig_import_helper

61. return importlib.import_module('_pywrap_tensorflow_internal')

62. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in import_module

63. return _bootstrap._gcd_import(name[level:], package, level)

64. ImportError: No module named '_pywrap_tensorflow_internal'

65.  
66.  

67. Failed to load the native TensorFlow runtime.

68.  

69. See https://www.tensorflow.org/install/install_sources#common_installation_problems

70.  

71. for some common reasons and solutions. Include the entire stack trace

72. above this error message when asking for help.

如果在 import tensorflow 的时候出现此问题，很有可能是没有装相应的vs运行时库，如tensorflow1.3，需要vs2015的运行时库即Visual C++ Redistributable for Visual Studio 2015

参考网址：http://blog.csdn.net/lwplwf/article/details/75175082

参考网址：http://blog.csdn.net/u010099080/article/details/53418159

参考网址：https://stackoverflow.com/questions/43942185/failed-to-load-the-native-tensorflow-runtime-python-3-5-2