tensorflow老版本的安装_TensorFlow 安装GPU版本

TensorFlow 有两个版本：CPU 版本和 GPU 版本。GPU 版本需要 CUDA 和 cuDNN 的支持，CPU 版本不需要。如果你要安装 GPU 版本，请先确认你的显卡支持 CUDA。我安装的是 GPU 版本，采用 pip 安装方式，所以就以 GPU 安装为例，CPU 版本只不过不需要安装 CUDA 和 cuDNN。

1. 确认你的显卡支持 CUDA

确认显卡是否支持

我这里是 GXT 1060

2.下载并安装 Visual Studio ，因为CUDA需要Visual Studio的支持。

image.png

安装选项：

image.png

3. 下载安装 CUDA Toolkit 9.0

下载：

image.png

安装：

image.png

4. 下载 cuDNN，并解压到目录，把对应的目录地址添加到path环境变量

image.png

下载 cuDNN必须需要一个账号，如果没有，注册一个新的账号，登录后才能下载。

下载：

image.png

解压

image.png

添加到环境变量

添加到环境变量

5. 安装 Anaconda，因为这个集成了很多科学计算所必需的库，能够避免很多依赖问题

image.png

把 Anaconda安装目录 添加到环境变量

image.png

6. 使用 pip 安装 tensorflow GPU版本

管理员身份打开cmd

image.png

输入：

pip install --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

7. 测试程序：

importtensorflowastfa = tf.random_normal((100,100))b = tf.random_normal((100,500))c = tf.matmul(a, b)#矩阵相乘sess = tf.InteractiveSession()sess.run(c)

执行结果

出现以上结果，说明安装成功了。