码迷，mamicode.com
首页 > Windows程序 > 详细

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

时间：2017-12-17 00:17:39      阅读：308      评论：0      收藏：0      [点我收藏+]

标签：解压   his   tools   .dll   pytho   session   tool   att   exception   

前言：

　　　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

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

标签：解压   his   tools   .dll   pytho   session   tool   att   exception   

原文地址：http://www.cnblogs.com/huaxingtianxia/p/8048018.html
踩
(0)
赞
(0)
   
举报
评论 一句话评论（0）
登录后才能评论！
分享档案
更多>
2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)
周排行
mamicode.com排行更多图片
更多
友情链接
兰亭集智   国之画   百度统计   站长统计   阿里云   chrome插件   新版天听网
关于我们 - 联系我们 - 留言反馈
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码！