Win10下安装并使用tensorflow-gpu1.8.0+python3.6全过程(显卡MX250+CUDA9.0+cudnn)

本文详细记录了在Win10系统中,安装tensorflow-gpu1.8.0、CUDA9.0和cudnn的过程,包括检查显卡信息、选择合适的CUDA版本、安装步骤以及最后的测试验证。在GPU(MX250)上运行,显著提升了训练速度。
摘要由CSDN通过智能技术生成

-----最近从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是向下兼容的)

在这里插入图片描述

下载完成后按照默认的执行下去(当然也可以修改安装的路径)
在安装时如果电脑装有vs2017,那么这里建议取消VS Integration
在这里插入图片描述

然后一步步往下执行
在这里插入图片描述

  • 检查cuda是否安装成功:输入nvcc --version

  • 1
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值