Win10下安装并使用下安装并使用tensorflow-gpu1.8.0+python3.6全过程(显卡全过程(显卡
MX250+CUDA9.0+cudnn))
—–最近从github上找了一个代码跑,但是cpu训练的时间实在是太长,所以想用gpu训练一下,经过了一天的折腾终于可以用gpu进行训练了,嘿嘿~
首先先看一下自己电脑的显卡信息:首先先看一下自己电脑的显卡信息:
可以看到我的显卡为MX250
然后进入然后进入NVIDIA控制面板控制面板->系统信息系统信息->组件组件 查看可以使用的查看可以使用的cuda版本版本
这里我先下载了cuda10.1的版本,不过后来我发现tensorflow-gpu 1.8.0仅支持cuda9.0的版本,所以之后我又重装了一遍cuda9.0,中间还经历了删除cuda10.0,两个版本的安装都是
一样的。
进入官网:进入官网:https://developer.nvidia.com/cuda-toolkit-archive
找到和显卡信息相匹配的找到和显卡信息相匹配的cuda((cuda是向下兼容的)是向下兼容的)
评论0