tensorflow-gpu 安装 tensorflow1.12 + CUDA 9.0 + cudnn 7.5 同时安装gpu版本与cpu版本

现在 windows10 支持 pip install tensorflow-gpu 直接安装

一、安装tensorflow-gpu

但是需要注意：

1、电脑要支持 CUDA（检查自己的GPU看支持到哪个版本，GTX960M支持9.0）

2、需要下载cudnn,这个深度学习加速库，并把cudnn下面每一项放到CUDA对应目录

 

可能出现的问题：

1、找不到DLL的情况

这个问题可能是，你没有把cudnn对应文件正确的放入CUDA/v9.0/xx目录下面。

也可能是你的CUDA与cudnn版本不兼容

还有可能就是你所pip安装的tensorflow和你安装的CUDA版本不兼容（因为pip默认安装当下最新tensorflow，然而他并不一点和你所安装的CUDA版本想兼容，例如我之前安装的tensorflow1.13，他和我安装的CUDA9.0不兼容，所以需要安装tensorflow1.12才行）

2、至于出现了一些警告：

可以不用管，也可以通过忽略解决。

 

二、安装tensorflow -cpu

1、需要另外设置一个虚拟环境进行安装，同时尽量保持 python版本一致，tensorflow版本一致

2、pip安装方式默认安装的是manylinux1_X86这是一个通用版本，在win10上面会有警告。

Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2

可直接到http://mirrors.aliyun.com/pypi/simple/tensorflow/下载amd_64版本whl文件安装。

3、若安装whl文件时候报错zipfile.BadZipFile: File is not a zip file，则说明安装包下载过程出错，重新下载即可。

 

查看GPU是否支持CUDA（计算能力大于3.1即可）: https://developer.nvidia.com/cuda-gpus

tensorflow与CUDA版本对应以及镜像源：https://blog.csdn.net/yuejisuo1948/article/details/81043962

安装步骤： https://www.jianshu.com/p/aafd88033867