安装tensorflow GPU

时间:2017-10-07 15:21:36

标签: python tensorflow tensorflow-gpu

我正在尝试在win10上安装tensorflow gpu。 我的步骤:

  1. 安装Python 3.5.2
  2. 安装Nvidia CUDA(v9)+ cudnn
  3. 在prog.files目录上复制cudnn文件NVIDIA GPU Computing Toolkit
  4. 安装tensorflow gpu(pip3 install --upgrade tensorflow-gpu)

    C:\的Python>蟒     win32上的Python 3.5.2(v3.5.2:4def2a2901a5,2016年6月25日,22:18:55)[MSC v.1900 64位(AMD64)]     输入“帮助”,“版权”,“信用”或“许可”以获取更多信息。

      
        
          

    将tensorflow导入为tf           Traceback(最近一次调用最后一次):             swig_import_helper中的文件“C:\ Python \ lib \ site-packages \ tensorflow \ python \ pywrap_tensorflow_internal.py”,第18行               return importlib.import_module(mname)             import_module中的文件“C:\ Python \ lib \ importlib__init __。py”,第126行               return _bootstrap._gcd_import(name [level:],package,level)             文件“”,第986行,在_gcd_import中             文件“”,第969行,在_find_and_load中             文件“”,第958行,在_find_and_load_unlocked中             文件“”,第666行,在_load_unlocked中             在module_from_spec中的文件“”,第577行             在create_module中输入文件“”,第906行             文件“”,第222行,在_call_with_frames_removed中           ImportError:DLL加载失败:Ненайденуказанныймодуль。

        
      
    During handling of the above exception, another exception occurred:
    
    Traceback (most recent call last):
      File "C:\Python\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 41, in <module>
        from tensorflow.python.pywrap_tensorflow_internal import *
      File "C:\Python\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 21, in <module>
        _pywrap_tensorflow_internal = swig_import_helper()
      File "C:\Python\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 20, in swig_import_helper
        return importlib.import_module('_pywrap_tensorflow_internal')
      File "C:\Python\lib\importlib\__init__.py", line 126, in import_module
        return _bootstrap._gcd_import(name[level:], package, level)
    ImportError: No module named '_pywrap_tensorflow_internal'
    
    During handling of the above exception, another exception occurred:
    
    Traceback (most recent call last):
      File "<stdin>", line 1, in <module>
      File "C:\Python\lib\site-packages\tensorflow\__init__.py", line 24, in <module>
        from tensorflow.python import *
      File "C:\Python\lib\site-packages\tensorflow\python\__init__.py", line 49, in <module>
        from tensorflow.python import pywrap_tensorflow
      File "C:\Python\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 52, in <module>
        raise ImportError(msg)
    ImportError: Traceback (most recent call last):
      File "C:\Python\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in swig_import_helper
        return importlib.import_module(mname)
      File "C:\Python\lib\importlib\__init__.py", line 126, in import_module
        return _bootstrap._gcd_import(name[level:], package, level)
      File "<frozen importlib._bootstrap>", line 986, in _gcd_import
      File "<frozen importlib._bootstrap>", line 969, in _find_and_load
      File "<frozen importlib._bootstrap>", line 958, in _find_and_load_unlocked
      File "<frozen importlib._bootstrap>", line 666, in _load_unlocked
      File "<frozen importlib._bootstrap>", line 577, in module_from_spec
      File "<frozen importlib._bootstrap_external>", line 906, in create_module
      File "<frozen importlib._bootstrap>", line 222, in _call_with_frames_removed
    ImportError: DLL load failed: Не найден указанный модуль.
    
    During handling of the above exception, another exception occurred:
    
    Traceback (most recent call last):
      File "C:\Python\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 41, in <module>
        from tensorflow.python.pywrap_tensorflow_internal import *
      File "C:\Python\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 21, in <module>
        _pywrap_tensorflow_internal = swig_import_helper()
      File "C:\Python\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 20, in swig_import_helper
        return importlib.import_module('_pywrap_tensorflow_internal')
      File "C:\Python\lib\importlib\__init__.py", line 126, in import_module
        return _bootstrap._gcd_import(name[level:], package, level)
    ImportError: No module named '_pywrap_tensorflow_internal'
    
    
    Failed to load the native TensorFlow runtime.
    
    See https://www.tensorflow.org/install/install_sources#common_installation_problems
    
    for some common reasons and solutions.  Include the entire stack trace
    above this error message when asking for help.
    
  5. 如果我在没有GPU的情况下安装tensorflow-一切正常。

7 个答案:

答案 0 :(得分:2)

遵循TensorFlow install instructions非常重要,从而选择正确的版本:

  
      
  • CUDA®Toolkit8.0 。有关详细信息,请参阅NVIDIA的文档确保将相关的Cuda路径名附加到%PATH%环境变量,如NVIDIA文档中所述。
  •   
  • 与CUDA Toolkit 8.0相关联的NVIDIA驱动程序。
  •   
  • cuDNN v6或v6.1 。有关详细信息,请参阅NVIDIA的文档。请注意,cuDNN通常安装在与其他CUDA DLL不同的位置。确保将安装cuDNN DLL的目录添加到%PATH%环境变量。
  •   
  • 具有CUDA Compute Capability 3.0或更高版本的GPU卡。有关支持的GPU卡列表,请参阅NVIDIA文档。
  •   

目前,CUDA Toolkit的最新版本是9.0,但对于Tensorflow,您需要版本8,可在CUDA toolkit archive上找到: https://developer.nvidia.com/cuda-80-ga2-download-archive

对于cuDNN(深度神经网络库),您需要选择 cuDNN v6 ,而不是最新版本v7:https://developer.nvidia.com/rdp/cudnn-download

最后,请务必将Windows %PATH%变量放入CUDA bin目录(dll库所在的目录)。

然后,您可以安装tensorflow:

pip3 install --upgrade tensorflow-gpu

如果它还不起作用,您可能需要先重启。

答案 1 :(得分:0)

目前,支持这些版本。不要尝试安装更新的版本。

Visual Studio 2015。 CUDA 9.0。 cuDNN 7.0(确保将dll复制到正确的目录)。 Python 3.5。

然后你可以安装tf

答案 2 :(得分:0)

这就是我的解决方法, 如果您使用的是pip或pip3,则第一次成功的机会就较小。

  • 卸载Python,CUDA,CuNN,Tensorflow-gpu(如果已安装)
  • 安装ANACONDA或MINICONDA
  • 打开ANACONDA提示符并运行以下命令:

    conda create --name tf_gpu tensorflow-gpu

  • 这将创建一个环境tf_gpu,其中将安装所有兼容版本的Python,CUDA,CuNN和Tensorflow

  • 在安装所有软件包后,打开ANACONDA提示符并键入以下命令

    conda激活tf_gpu

  • 这会将conda'(base)'更改为'(tf_gpu)'环境

  • 然后像往常一样键入python以激活python提示符

  • 如果要停用'tf_gpu'环境

    停用conda

答案 3 :(得分:0)

我自己尝试了很多博客。我总是遇到一个或其他错误。 但是,以下对我来说非常有效。

conda create --name tf_gpu tensorflow-gpu 

您只需要在您的系统上安装 Nvidia 驱动程序。无需安装CUDA和CudNN,也无需担心版本兼容性。

此命令将为您安装 CUDA 和 CudNN 以及兼容版本的 Tensorflow。

答案 4 :(得分:0)

最后这个文档对我有用:Anaconda | TensorFlow

该文件是不言自明的。如果您已经有环境,只需运行:

conda install tensorflow-gpu 

重要说明:

  • GPU TensorFlow 使用 CUDA。
  • 在 Windows 和 Linux 上,TensorFlow 2.0 版本仅支持 CUDA 10.0。以前版本的 TensorFlow 支持其他版本的 CUDA。

答案 5 :(得分:0)

好的,所以我参考了很多博客和 StackOverflow 的答案,但没有一个对我有用。但是,我终于找到了一个对我来说非常有效的程序。所以,我有一个 NVIDIA GTX 1650 GPU,以下是我为使启用 GPU 的 TensorFlow 启动并运行而遵循的步骤:

1) 安装 NVIDIA 驱动程序(Studio 驱动程序)并重新启动您的 PC。

2) 下载 Visual Studio 2019 社区版(在其中安装所有 c++、python/AI/DL 依赖项)。

3) 安装 Cuda 10.1(更新 2,2019 年 8 月发布)。

4) 安装 CuDNN 7.6.x(相应地选择 x,因此根据 Cuda 10.1(2019 年 8 月),x 的适当值为 3)。所以我基本上为 Cuda 10.1 安装了 CuDNN 7.6.3

5) 然后,我们必须将Cuda的bin、include、lib\x64文件夹添加到系统路径中。 (注意include,安装Cuda后默认会在路径中添加lib\x64,所以我们只需要在路径中添加Cuda的bin文件夹即可)

6) 接下来,我们必须将 CuDNN 的 bin 文件夹添加到系统路径中。

7) 最后,在 CMD 中输入 pip install TensorFlow,然后重启你的电脑。

请注意如果您有不同的GPU,您可以查看Cuda和CuDNN的官方网站,以查看与您的GPU兼容的Cuda和CuDNN版本。

答案 6 :(得分:-1)

如果您使用的是anaconda,请在anaconda提示符中运行此命令

$fields['main_img'] = BaseFieldDefinition::create('image') ->setLabel(t('Main image of the hardware')) ->setSettings([ 'file_directory' => 'hardware', 'file_extensions' => 'png jpg jpeg', ]) ->setDisplayOptions('view', array( 'label' => 'above', 'type' => 'image', 'weight' => -30, )) ->setDisplayOptions('form', array( 'label' => 'above', 'type' => 'image_image', 'weight' => -30, )) ->setDisplayConfigurable('form', TRUE) ->setDisplayConfigurable('view', TRUE);

这也将为您安装conda install -c anaconda tensorflow-gpucuda toolkit,您很高兴  在网站上它提到这个命令是针对linux的,所以我不确定它是否适用于窗口,但你可以试试