Windows 下安装 TensorFlow-gpu 教程

最新推荐文章于 2024-08-13 13:48:39 发布

Wonz

最新推荐文章于 2024-08-13 13:48:39 发布

阅读量764

点赞数

分类专栏： # 软件安装文章标签： TensorFlow Windows 深度学习 gpu CUDA

本文链接：https://blog.csdn.net/wonz5130/article/details/84073782

版权

软件安装专栏收录该内容

34 篇文章 3 订阅

订阅专栏

终于迈开了这一步，TensorFlow 大法好。

一、介绍

TensorFlow™ 是一个采用数据流图（data flow graphs），用于数值计算的开源软件库。节点（Nodes）在图中表示数学操作，图中的线（edges）则表示在节点间相互联系的多维数据数组，即张量（tensor）。它灵活的架构让你可以在多种平台上展开计算，例如台式计算机中的一个或多个 CPU（或 GPU），服务器，移动设备等等。TensorFlow 最初由 Google 大脑小组（隶属于 Google 机器智能研究机构）的研究员和工程师们开发出来，用于机器学习和深度神经网络方面的研究，但这个系统的通用性使其也可广泛用于其他计算领域。（来自TensorFlow中文社区）
在这里插入图片描述

一开始还没发现这图标的意思，直到偶然间，发现正好是 T 和 F 的影子重叠在一起构成的。TensorFlow 的 LOGO 是真的好看。

二、安装 TensorFlow

我是直接用 pip 安装的 GPU 版本。

pip3 install --upgrade tensorflow-gpu

这个安装好之后，还不能跑 TensorFlow ，还缺两个东西。

三、安装 CUDA

这个你要确定版本，非常重要！否则不能使用。可以先找到对应的显卡：网址在这里。我的是 GTX960M 的显卡。

我当时按照上面的指令安装 TensorFlow 的时候，自动装的是比较新的版本。然后我就去装了最新的 CUDA ，结果发现不能用，又试了几个版本，最终装了 CUDA 9.0 才成功。

先去官网找到对应的版本。
在这里插入图片描述

再选择相应的信息，点击 Download 即可。
在这里插入图片描述

下载完后，直接点击 exe 进行安装。下面演示的是 CUDA 10 安装过程。
在这里插入图片描述

选择 精简 安装。

在这里插入图片描述

四、安装 cuDNN

首先，要去官网注册个账号。然后，选择与 CUDA 相对应的版本。
在这里插入图片描述

下载完解压，我是把 cuda 文件夹解压到 C盘 的。
在这里插入图片描述

添加环境变量。
在这里插入图片描述

五、测试

import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)