用pip安装Tensorflow

时间:2017-07-10 16:46:45

标签: python tensorflow pip

我已经用pip安装了tensorflow

pip3 install --upgrade tensorflow-gpu

然后,当我在python中时,我尝试

import tensorflow

我收到this 错误。 https://i.gyazo.com/0a66a7ceef62a8de55a4d59c58e57388.png

我有最新版本的python和windows 10 x64

1 个答案:

答案 0 :(得分:0)

屏幕截图中的错误始于DLL加载失败 - 我注意到您使用pip3安装并且安装在Windows上安装以供GPU使用。 This guide以您可能忽略的先决条件开头:

  

在Windows上安装TensorFlow

     

本指南介绍如何在Windows上安装TensorFlow。

     

确定要安装哪个TensorFlow

     

您必须选择以下类型的TensorFlow才能安装:

     

TensorFlow仅支持CPU。 如果您的系统没有   NVIDIA®GPU,您必须安装此版本。请注意此版本的   TensorFlow通常更容易安装(通常在5或10中   分钟),所以即使你有NVIDIA GPU, 我们也建议安装   这个版本首先。

     

支持GPU的TensorFlow。 TensorFlow程序   通常在GPU上比在CPU上运行速度快得多。因此,   如果您的系统具有符合以下所示先决条件的NVIDIA®GPU   你需要运行性能关键型应用程序   最终安装此版本。

     

使用GPU支持运行TensorFlow的要求

     

如果您正在使用其中一个安装支持GPU的TensorFlow   本指南中描述的机制,然后是以下NVIDIA软件   必须安装在您的系统上:

     

CUDA®Toolkit8.0。有关详细信息,请参阅NVIDIA的文档确保   您将相关的Cuda路径名附加到%PATH%环境   NVIDIA文档中描述的变量。 NVIDIA驱动程序   与CUDA Toolkit 8.0相关联。 cuDNN v5.1。有关详情,请参阅   NVIDIA的文档。请注意,cuDNN通常安装在   与其他CUDA DLL不同的位置。 确保添加   将cuDNN DLL安装到%PATH%环境的目录   变量即可。具有CUDA Compute Capability 3.0或更高版本的GPU卡。看到   NVIDIA文档,列出支持的GPU卡列表。如果你有   前面的软件包的早期版本,请升级到   指定版本。

以上链接页面,但添加了我的突出显示,(以及其他样式丢失)。

您还需要确保机器上安装了兼容的图形卡。如果您有任何疑问,请尝试:

pip3 install --upgrade  tensorflow

您将获得基于CPU的tensorflow版本,无需安装外部软件。在路径上,但会运行得更慢。