win10+tensorflow-gpu1.12.0+python3.6.0+cuda9.0.176+cudnn7.5.0环境配置

此贴为了记录自己在环境配置时遇到的坑,同时为其他想配置环境的人提供参考

 最近在搞些NLP的东西,用电脑CPU训练数据时,发现太慢,后来想用GPU训练数据,便将CPU版本卸载,试着去安装GPU版本的Tensorflow。(其中的版本对应关系来自此博客

1、下载标题中的各个版本软件,因为tensorflow-gpu每个版本都有对应版本的cuda和cudnn,所以在下载时请务必下载正确。

2、cuda的安装参考此博客,我参考的安装部分,后面的环境变量不用配置,因为在安装的时候默认会自动设置环境变量。

3、cudnn的下载链接。点进去发现各个版本的cudnn,下载自己需要的。

win10+tensorflow-gpu1.12.0+python3.6.0+cuda9.0.176+cudnn7.5.0环境配置

下载之后解压,将对应文件夹下的文件复制到cuda对应的文件夹下,另外两个文件相同操作。

win10+tensorflow-gpu1.12.0+python3.6.0+cuda9.0.176+cudnn7.5.0环境配置

4、安装完成之后,进行测试。可能会出现

from google.protobuf.pyext import _message

这一错误。如若出现,则是protobuf的版本不正确的问题。将protobuf的版本更新至3.6.0.如果还未解决,请搜索对应正确的版本。(我自己的是下载的3.6.0解决了这个问题)