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版本(版本一定不要搞错)下载地址点击下载
下载完之后在安装,我在安装时360管家一直说有病毒阻止安装过程,所以在安装时候把360管家退了。除了这个应该基本可以安装成功。
tensorflow-gpu==1.11.0安装所需cuda9.0:
下载cuda9.0版本(版本一定不要搞错)下载地址点击下载
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相应的目录中:
复制 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内容就可以,下载的很慢,等吧
6. python中验证是否成功
这就表示安装成功
7.pycharm中导入tensorflow-gpu
(1)找到python安装的位置,在cmd中输入,如下
(2)打开pycharm---->file----->setting----->project interpreter----->点击红圈中的add---->再按照第二张图中的顺序插入python安装路径就行了。
(3)如果出现下图就表示导入成功