WINDOWS10 Installing the TensorFlow GPU version (PIP3 installation method)

Objective:

TensorFlow has two versions of CPU and GPU: GPU version requires NVIDIA Cuda and CuDNN support, CPU version is not required; This article mainly installs the GPU version.

1. Environment

• GPU: Verify that your video card supports CUDA, which is confirmed here.
• VS2015 Runtime Library: Download 64-bit, download here, install after download.
• Python 3.6/3.5: Download 64-bit, download here, install after download.
• Pip 9.0.1 (confirm pip version >= 8.1, use pip-v to view the current PIP version, and upgrade Pip with python-m pip install-u pip. ）

Note: Pip actually comes in when you install Python, but it needs to be (xxx:\ python\python36\scripts) is configured in the environment variable (path), the red font represents the path to the Python installation.

2. Installation

(1), download CUDA
Google official recommended CUDA8, can go to the official website download: Cuda

• Cuda Installation

Directly double-click the EXE installation, follow the prompts to install the end of the line. After the test is installed, the command line enters Nvcc-v and the version information indicates that the installation was successful.

(2), download CUDNN

TensorFlow version different, the need for the CUDNN version is not the same, see TensorFlow release notes, such as: tensorflow1.3 Release Notes

• Configure CUDNN

Download to the corresponding version of CUDNN (tensorflow1.3 need cuDNN6, can be downloaded to https://www.zhihu.com/question/37082272), unzip:

1. The extracted bin directory is configured into the system environment variable (path);
2. Copy the appropriate files to the appropriate directory:

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\

(3), installation TensorFlow

[HTML]View PlainCopy

1. # GPU Version
2. PIP3 Install--upgrade Tensorflow-gpu
4. # CPU Version
5. PIP3 Install--upgrade TensorFlow

(4), uninstall TensorFlow

[HTML]View PlainCopy

1. PIP3 Uninstall TensorFlow

3. Testing [HTML]View PlainCopy

1. Import TensorFlow as TF
3. A = Tf.random_normal ((+))
4. b = Tf.random_normal ((+, +))
5. c = Tf.matmul (A, b)
6. Sess = TF. InteractiveSession ()
7. Sess.run (c)

Issue [HTML]View PlainCopy

2. Traceback (most recent):
4. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_ tensorflow_internal.py ", line +, in Swig_import_helper
6. Return Importlib.import_module (Mname)
8. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in Import_module
10. Return _bootstrap._gcd_import (name[level:], package, level)
12. File "<frozen Importlib._bootstrap>", line 986, in _gcd_import
14. File "<frozen Importlib._bootstrap>", Line 969, in _find_and_load
16. File "<frozen Importlib._bootstrap>", Line 958, in _find_and_load_unlocked
18. File "<frozen Importlib._bootstrap>", Line 666, in _load_unlocked
20. File "<frozen Importlib._bootstrap>", Line 577, in Module_from_spec
22. File "<frozen importlib._bootstrap_external>", Line 906, in Create_module
24. File "<frozen Importlib._bootstrap>", Line 222, in _call_with_frames_removed
26. Importerror:dll load failed:the specified module could not be found.
30. During handling of the above exception, another exception occurred:
34. Traceback (most recent):
36. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_ tensorflow.py ", line $, in <module>
38. From tensorflow.python.pywrap_tensorflow_internal Import *
40. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_ tensorflow_internal.py ", line +, in <module>
42. _pywrap_tensorflow_internal = swig_import_helper ()
44. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_ tensorflow_internal.py ", line A, in Swig_import_helper
46. Return Importlib.import_module (' _pywrap_tensorflow_internal ')
48. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in Import_module
50. Return _bootstrap._gcd_import (name[level:], package, level)
52. Importerror:no module named ' _pywrap_tensorflow_internal '
56. During handling of the above exception, another exception occurred:
60. Traceback (most recent):
62. File "<pyshell#0>", Line 1, in <module>
64. Import TensorFlow as TF
66. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\__init__.py", line +, in <module>
68. From Tensorflow.python Import *
70. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\__init__.py", Line Wuyi, in <module>
72. From Tensorflow.python import Pywrap_tensorflow
74. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_ tensorflow.py ", line-in <module>
76. Raise Importerror (MSG)
78. Importerror:traceback (most recent):
80. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_ tensorflow_internal.py ", line +, in Swig_import_helper
82. Return Importlib.import_module (Mname)
84. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in Import_module
86. Return _bootstrap._gcd_import (name[level:], package, level)
88. File "<frozen Importlib._bootstrap>", line 986, in _gcd_import
90. File "<frozen Importlib._bootstrap>", Line 969, in _find_and_load
92. File "<frozen Importlib._bootstrap>", Line 958, in _find_and_load_unlocked
94. File "<frozen Importlib._bootstrap>", Line 666, in _load_unlocked
96. File "<frozen Importlib._bootstrap>", Line 577, in Module_from_spec
98. File "<frozen importlib._bootstrap_external>", Line 906, in Create_module
100. File "<frozen Importlib._bootstrap>", Line 222, in _call_with_frames_removed
102. Importerror:dll load failed:the specified module could not be found.
106. During handling of the above exception, another exception occurred:
110. Traceback (most recent):
112. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_ tensorflow.py ", line $, in <module>
114. From tensorflow.python.pywrap_tensorflow_internal Import *
116. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_ tensorflow_internal.py ", line +, in <module>
118. _pywrap_tensorflow_internal = swig_import_helper ()
120. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\pywrap_ tensorflow_internal.py ", line A, in Swig_import_helper
122. Return Importlib.import_module (' _pywrap_tensorflow_internal ')
124. File "C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\importlib\__init__.py", line 126, in Import_module
126. Return _bootstrap._gcd_import (name[level:], package, level)
128. Importerror:no module named ' _pywrap_tensorflow_internal '
134. Failed to load the native TensorFlow runtime.
138. See Https://www.tensorflow.org/install/install_sources#common_installation_problems
142. For some common reasons and solutions. Include the entire stack trace
144. Above this error message is asking for help.

If the Import TensorFlowWhen this problem occurs, it is most likely that the corresponding VS runtime library, such as tensorflow1.3, is not installed, and the runtime library that requires vs2015 is Visual C + + Redistributable for Visual Studio 2015

WINDOWS10 Installing the TensorFlow GPU version (PIP3 installation method)