zoukankan      html  css  js  c++  java

  • 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: PythonPython36Scripts）配置到环境变量(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 cudnnincudnn64_5.dll to C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0in
    　　　　copy cudnnincludecudnn.h to C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0include
    　　　　copy cudnnlibx64cudnn.lib to C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0libx64

    （３）、安装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:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpythonpywrap_tensorflow_internal.py", line 18, in swig_import_helper  
    3.     return importlib.import_module(mname)  
    4.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libimportlib\__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:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpythonpywrap_tensorflow.py", line 41, in <module>  
    19.     from tensorflow.python.pywrap_tensorflow_internal import *  
    20.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpythonpywrap_tensorflow_internal.py", line 21, in <module>  
    21.     _pywrap_tensorflow_internal = swig_import_helper()  
    22.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpythonpywrap_tensorflow_internal.py", line 20, in swig_import_helper  
    23.     return importlib.import_module('_pywrap_tensorflow_internal')  
    24.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libimportlib\__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:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflow\__init__.py", line 24, in <module>  
    34.     from tensorflow.python import *  
    35.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpython\__init__.py", line 51, in <module>  
    36.     from tensorflow.python import pywrap_tensorflow  
    37.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpythonpywrap_tensorflow.py", line 52, in <module>  
    38.     raise ImportError(msg)  
    39. ImportError: Traceback (most recent call last):  
    40.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpythonpywrap_tensorflow_internal.py", line 18, in swig_import_helper  
    41.     return importlib.import_module(mname)  
    42.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libimportlib\__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:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpythonpywrap_tensorflow.py", line 41, in <module>  
    57.     from tensorflow.python.pywrap_tensorflow_internal import *  
    58.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpythonpywrap_tensorflow_internal.py", line 21, in <module>  
    59.     _pywrap_tensorflow_internal = swig_import_helper()  
    60.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libsite-packages ensorflowpythonpywrap_tensorflow_internal.py", line 20, in swig_import_helper  
    61.     return importlib.import_module('_pywrap_tensorflow_internal')  
    62.   File "C:UsersAdminAppDataLocalProgramsPythonPython35libimportlib\__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
  • 相关阅读:
    python 闭包
    并行【parallel】和并发【concurrency】线程是并发还是并行，进程是并发还是并行
    生成器，送代器的区别
    深拷贝和浅拷贝的区别
    对缺省参数的理解
    对不定长参数的理解
    mysql三范式
    mysql基础查询语法
    mysql常见查询练习题
    Ubantu-Nginx部署
  • 原文地址：https://www.cnblogs.com/huaxingtianxia/p/8048018.html
Copyright © 2011-2022 走看看