使用 pip 安装 TensorFlow

Question

我已经用 pip 安装了 tensorflow

pip3 install --upgrade tensorflow-gpu

然后当我在 python 中并尝试

import tensorflow

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

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

Answer 1

您在屏幕截图中遇到的错误以 DLL Load failed 开头 - 我注意到您安装了 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® Toolkit 8.0。有关详细信息，请参阅 NVIDIA 的文档确保 您将相关的 Cuda 路径名附加到 %PATH% 环境 变量，如 NVIDIA 文档中所述。 NVIDIA 驱动程序 与 CUDA Toolkit 8.0 相关联。 cuDNN v5.1。有关详细信息，请参阅 英伟达的文档。请注意，cuDNN 通常安装在 与其他 CUDA DLL 不同的位置。 确保添加 将 cuDNN DLL 安装到 %PATH% 环境的目录 变量。具有 CUDA 计算能力 3.0 或更高版本的 GPU 卡。看 支持的 GPU 卡列表的 NVIDIA 文档。如果你有一个 上述软件包的早期版本，请升级到 指定的版本。

以上来自链接页面，但添加了我的突出显示，（以及其他样式丢失）。

您还需要确保机器上安装了兼容的显卡。如果您有任何疑问，请尝试：

pip3 install --upgrade  tensorflow

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