windows tensorflow-gpu+cuda+cudnn+python+pycharm

原来一直用的Ubuntu下的tensorflow-gpu,由于目前做一些工程上的应用,所以在windows安装tensorflow-gpu更方便使用。网上找了很多安装的方法错误很多,有时tensorflow版本不对,有时pycharm导入不了tensorflow-gpu,下面是我的安装经验,我都试了一遍,是可以用的,如果你有错误请在下面回复,也许能帮到你。

0. 安装tensorflow-gpu总体流程

(1)安装python3.6

(2)安装pycharm

(3)安装cuda(不同版本的tensorflow-gpu需要用到不同版本的cuda)

(4)安装cudnn(不同版本的tensorflow-gpu需要用到不同版本的cudnn)

(5)pycharm中导入tensorflow-gpu

1. 安装python3.6

安装python网上的资源很多,在百度链接很多,照着做就可以了,不难。

2.安装pycharm

安装pycharm网上的资源也很多,可以参考此链接

3.cuda的安装

网上比较多的是安装tensorflow-gpu==1.4.0版本的教程(在github上copy代码是要看清楚支持的tensorflow的版本,现在好多代码低版本的会报错,我就是因tensorflow-gpu==1.4.0与代码才换tensorflow-gpu==1.11.0),以下cuda安装都需要默认安装路径为了减少麻烦

tensorflow-gpu==1.4.0安装所需cuda8.0:

下载cuda8.0版本(版本一定不要搞错)下载地址点击下载

windows tensorflow-gpu+cuda+cudnn+python+pycharm

下载完之后在安装,我在安装时360管家一直说有病毒阻止安装过程,所以在安装时候把360管家退了。除了这个应该基本可以安装成功。

tensorflow-gpu==1.11.0安装所需cuda9.0:

下载cuda9.0版本(版本一定不要搞错)下载地址点击下载

windows tensorflow-gpu+cuda+cudnn+python+pycharm

4.cudnn的安装

同样,不同的tensorflow-gpu版本需要不同的cudnn版本。

tensorflow-gpu==1.4.0+cuda8.0需要cudnn6:

官网下载比较麻烦,可以直接在这里下载cudnn6点此下载。下载后解压。解压过后将相应的文件拷贝到cuda8.0相应的目录中:

复制 cudnn\bin\cudnn64_5.dll 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\
复制 cudnn\include\cudnn.h    到  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include\
复制 cudnn\lib\x64\cudnn.lib   到  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64\

tensorflow-gpu==1.11.0+cuda9.0需要cudnn7.5:

可以直接在这里下载cudnn7.5,微信可以直接登录。下载后解压。解压过后将相应的文件拷贝到cuda9.0相应的目录中:

windows tensorflow-gpu+cuda+cudnn+python+pycharm

复制 cudnn\bin\cudnn64_5.dll 到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\
复制 cudnn\include\cudnn.h    到  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include\
复制 cudnn\lib\x64\cudnn.lib   到  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64\

5. tensorflow-gpu安装下载

进入windows中的cmd,如果用的是tensorflo-gpu==1.4.0输入pip install tensorflo-gpu==1.4.0内容就可以,下载的很慢,如果用的是tensorflo-gpu==1.11.0输入pip install tensorflo-gpu==1.11.0内容就可以,下载的很慢,等吧

windows tensorflow-gpu+cuda+cudnn+python+pycharm

6. python中验证是否成功

这就表示安装成功

windows tensorflow-gpu+cuda+cudnn+python+pycharm

7.pycharm中导入tensorflow-gpu

(1)找到python安装的位置,在cmd中输入,如下

windows tensorflow-gpu+cuda+cudnn+python+pycharm

(2)打开pycharm---->file----->setting----->project interpreter----->点击红圈中的add---->再按照第二张图中的顺序插入python安装路径就行了。

windows tensorflow-gpu+cuda+cudnn+python+pycharm

windows tensorflow-gpu+cuda+cudnn+python+pycharm

(3)如果出现下图就表示导入成功

windows tensorflow-gpu+cuda+cudnn+python+pycharm