最近要用到tensorflow,之前安装了cpu版本的,但训练模型速度实在是慢,打算安装一个GPU版本的tensorflow:

安装环境为:wiindows 10, anaconda python36, pycharm

tensorflow版本与cuda, cudnn版本的对应关系:

   tensorflow-gpu v1.9.0 | cuda9.0 |  cuDNN7.1.4可行  | 备注:7.0.4/ 7.0.5/ 7.1.2不明确

   tensorflow-gpu v1.8.0 | cuda9.0 |  cuDNN  不明确 | 备注:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4

   tensorflow-gpu v1.7.0 | cuda9.0 |  cuDNN  不明确 | 备注:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4

   tensorflow-gpu v1.6.0 | cuda9.0 |  cuDNN  不明确 | 备注:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4

   tensorflow-gpu v1.5.0 | cuda9.0 |  cuDNN  不明确 | 备注:7.0.4/ 7.0.5/ 7.1.2/ 7.1.4

   tensorflow-gpu v1.4.0 | cuda8.0 |  cuDNN 6.0 | 备注:6.0正常使用, 7.0.5不能用,5.1未知 

   tensorflow-gpu v1.3.0 | cuda8.0 |  cuDNN 6.0 | 备注:6.0正常使用, 7.0.5不能用,5.1未知 

   tensorflow-gpu v1.2.0 | cuda8.0 |  cuDNN 5.1 | 备注:5.1正常使用, 6.0/ 7.0.5 未知

   tensorflow-gpu v1.1.0 | cuda8.0 |  cuDNN 5.1 | 备注:5.1正常使用, 6.0/ 7.0.5 未知                    (参考

请严格按照上面的版本安装,安装顺序为visual studio(关于visual studio的安装就不赘述), cuda, cudnn,tensorflow,严格遵守才能少踩坑,我选择安装cuda8.0, cudnn6.0, tensorflow1.3.0

1. 先下载CUDA 8.0,按照官方文档的要求,否则会出问题:CUDA8.0下载地址,选择本地下载总是在99%的时候下载失败,1.2G的东西这样真的是坑人,所以最后选择在线安装, 有大佬知道原因请告知我一下。。。。如果在之前安装了错误的版本,请卸载干净后再安装

TensorFlow训练时候如何使用共享gpu内存 tensorflow配置gpu_CUDA

2.配置CUDA环境变量:

tensorflow是一个编程系统,它以计算图(graph)的形式来表示各种运算之间的关系,它所有的运算操作都必须在会话(session)中完成。在运行tf.session()的时候会遇到各种lib, dll文件加载不了的问题,是因为在session()在运行时调用CUDA,而CUDA的环境变量配置有问题,就会出现各种报错。

安装完cuda后,打开环境变量,发现已有两个CUDA_PATH和CUDA_PATH_V8_0两个环境变量。

TensorFlow训练时候如何使用共享gpu内存 tensorflow配置gpu_CUDA_02

需要再添加两个环境变量才可以:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64

添加完环境变量后:

TensorFlow训练时候如何使用共享gpu内存 tensorflow配置gpu_tensorflow_03

最后在cmd里输入 echo %path% 就能查看你的是否添加进环境变量了:

TensorFlow训练时候如何使用共享gpu内存 tensorflow配置gpu_CUDA_04

3.安装cudnn   cudnn6.0下载.,注册账号,耐心填完问卷即可下载

解压下载文件后可以看到cudnn文件夹如下:

TensorFlow训练时候如何使用共享gpu内存 tensorflow配置gpu_环境变量_05

打开cuda的安装目录发现也有这样的三个文件夹

TensorFlow训练时候如何使用共享gpu内存 tensorflow配置gpu_tensorflow_06

只需将cudnn三个文件夹下的内容复制到cuda对应的文件夹下,出现替换时进行替换。

4.安装tensorflow_GPU

(1)在pycharm环境下安装可以通过file->settings->project intepreter,安装,点击左侧栏+,在搜索栏输入tensorflow_gpu即可,

        但这种方法安装速度极慢。

(2)推荐方法, 通过anacond prompt安装,采用国内的镜像源,安装速度大大提升。关于镜像源安装方法。请参考

         这篇博客 ,采用如下命令即可安装

         pip install tensorflow_gpu -i http://pypi.mirrors.ustc.edu.cn/simple --trusted-host pypi.mirrors.ustc.edu.cn

        这条命令默认安装的是最新版本,所以我们需要做一下修改

        pip install tensorflow_gpu==1.3.0 -i http://pypi.mirrors.ustc.edu.cn/simple --trusted-host pypi.mirrors.ustc.edu.cn

        安装完成后即可使用。

5.运行一个测试程序:

  

TensorFlow训练时候如何使用共享gpu内存 tensorflow配置gpu_tensorflow_07

   安装成功~