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,下载自己需要的。

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

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

from google.protobuf.pyext import _message

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值