繁体   English   中英

用pip安装Tensorflow

[英]installing Tensorflow with pip

我已经用pip安装了tensorflow

pip3 install --upgrade tensorflow-gpu

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

import tensorflow

我得到这个错误。 https://i.gyazo.com/0a66a7ceef62a8de55a4d59c58e57388.png

我拥有最新版本的python和Windows 10 x64

屏幕快照中出现的错误以DLL Load开头失败-我注意到您已安装pip3,并且已安装在Windows上供GPU使用。 本指南从您可能已经忽略的先决条件开始:

在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®工具包8.0。 有关详细信息,请参阅NVIDIA文档。 确保按照 NVIDIA文档中的说明将相关的Cuda路径名附加到%PATH%环境变量中。 与CUDA Toolkit 8.0关联的NVIDIA驱动程序。 cuDNN v5.1。 有关详细信息,请参阅NVIDIA文档。 请注意,cuDNN通常安装在与其他CUDA DLL不同的位置。 确保将cuDNN DLL的安装目录添加到%PATH%环境变量中 具有CUDA计算能力3.0或更高版本的GPU卡。 有关受支持的GPU卡的列表,请参阅NVIDIA文档。 如果您具有上述软件包的早期版本,请升级到指定的版本。

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

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

pip3 install --upgrade  tensorflow

您将获得基于CPU的Tensorflow版本,该版本不需要在路径上安装外部软件,但运行速度较慢。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM