碼上歡樂
相關內容    簡體    繁體

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

本文轉載自   查看原文   2017-12-16 23:07   9440    tensorflow

前言：

　　　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

 

[html]  view plain  copy
 
  1. # GPU版本  
  2. pip3 install --upgrade tensorflow-gpu  
  3.   
  4. # CPU版本  
  5. pip3 install --upgrade tensorflow  

 

 

（４）、卸載tensorflow

 

[html]  view plain  copy
 
  1. pip3 uninstall tensorflow  

3、測試

[html]  view plain  copy
 
  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:

[html]  view plain  copy
 
  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


免責聲明！

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。



猜您在找 pip3安裝Tensorflow-GPU windows10如何安裝cpu版本tensorflow Windows10上安裝Keras 和 TensorFlow-GPU windows10下安裝tensorflow2.0-GPU和Cupy（不用搞CUDA+cudnn） TensorFlow-gpu運行問題記錄-windows10 在Windows 10 + Python 3.6.5 中用 pip 安裝最新版 TensorFlow v1.8 for GPU Windows下安裝TensorFlow GPU版本教程 Windows10中使用Anaconda安裝keras-gpu版本（遇到的坑） python2.7，python3.6，python3.8多版本windows10安裝，pip共存 Windows10中用Anaconda安裝tensorflow2.1.0(gpu版）（新手入門級）
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM