TensorFlow-软件安装-Windows下安装TensorFlow

0.声明:

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

本文参考了TensorFlow官方网站,仅供各位读者学习参考之用,https://www.tensorflow.org/install/install_windows?hl=zh-cn

1.硬件要求:

虽然这些说明可能也适用于其他 Windows 版本,但我们只在满足以下要求的计算机上验证过这些说明(而且我们只支持在此类计算机上按这些说明操作):

  • 64 位、x86 台式机或笔记本电脑
  • Windows 7 或更高版本

2.安装CUDA,cuDNN

注意:安装GPU版本的TensorFlow,必须具备NVIDIA显卡,并且安装CUDA工具包 9.0和cuDNN v7.0。并且,建议在安装之前,先确认安装了合适的显卡驱动。

  • CUDA® 工具包 9.0。官网下载安装即可,安装完成后,正常情况下,程序会自动添加环境。如果没有添加,请务必按照 NVIDIA 文档中的说明将相关的 CUDA 路径名附加到 %PATH% 环境变量上。
  • cuDNN v7.0。从官网下载,并且要与CUDA版本和系统版本对应。下载成功后,文件解压,找任意位置存放,然后将解压文件夹下的bin文件夹加入环境变量:

注意:如果您的某个软件包不同于上述版本,请改为指定的版本。特别是,cuDNN 版本必须完全匹配:如果无法找到 cuDNN64_7.dll,TensorFlow 就不会加载。要使用不同版本的 cuDNN,您必须从源代码构建。

3.使用原生 pip 进行安装TensorFlow

在 Windows 上,TensorFlow 支持 Python 3.5.x 和 3.6.x。请注意,Python 3 附带有 pip3 软件包管理器,您需使用此程序来安装 TensorFlow。

CPU版本安装:(如果您没有合适的显卡,可以考虑安装CPU版本。安装CPU版本,可忽略上述步骤2)

C:\> pip3 install --upgrade tensorflow

GPU版本安装:(需要成功安装CUDA和cuDNN,才能进行这一步。在强调一遍,一定要注意CUDA和cuDNN版本的一致性,强烈建议与官网推荐版本一致。本人曾在CUDA9.1版本下,成功安装了TensorFlow ,一般功能可正常使用,但运行CNN网络时,就一直报错了,无奈只能重装。

C:\> pip3 install --upgrade tensorflow-gpu

4.检查是否安装成功

在 Python 交互式 shell 中输入以下几行简短的程序代码:

5.官网安装文档

链接:https://pan.baidu.com/s/1Hgl6_flgUipKhpLN3qpqyA 密码:1wfn

猜你喜欢

转载自blog.csdn.net/qq_17753903/article/details/82597229
今日推荐