pip3安装Tensorflow-GPU

系统：win7

Python 3.6

安装方法：直接使用pip3 安装

官网上有详细的要求及安装方法：https://tensorflow.google.cn/install/gpu

首先，我们可以看到安装GPU版需要电脑软硬件满足一定的条件，即需要首先安装CUDA和cudnn, 这个直接点击蓝色的字体便能进入到下载界面，按顺序下载安装即可。

 

1） 下载安装CUDA

https://developer.nvidia.com/cuda,ANZHAO 选择自己需要的版本下载，下载完成后直接安装，安装步骤很简单，记一下安装路径，后面会用到。

 

2） 下载安装cudnn

https://developer.nvidia.com/cudnn

这个下载前需要先注册一个账户，但是也没有特殊要求，就简单注册一下就好了。

下载完成后解压文件（看一个帖子里说要把压缩包的扩展名改为.tgz格式在解压，我直接照做了），解压后的文件夹名为cuda,文件内容如下图

将其中的文件拷贝到先前cuda 安装路径下的对应文件夹中

 

3） 在环境变量中添加路径

将cuda安装目录下的bin和lib\x64两个文件夹的路径添加到环境变量PATH里面

4） pip3 安装tensorflow-gpu

打开cmd窗口，输入pip3 install tensorflow-gpu

安装成功后会显示 successfully

5） 安装完成，测试。

编写简单代码，测试tensorflow是否可以正常使用。

import tensorflow as tf

hello=tf.constant('hello')

sess=tf.Session()

print (sess.run(hello))

输出”hello”且未报错即安装成功。

此外，还可使用一下代码验证：

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

在结果中可以看到GPU, 即安装成功。