Win10下安装并使用tensorflow-gpu全过程

cpu训练的时间实在是太长,所以想用gpu训练一下,经过了一天的折腾终于可以用gpu进行训练了,嘿嘿~~~

  • 首先先看一下自己电脑的显卡信息,是否存在独显:

  • 然后进入NVIDIA控制面板->系统信息->组件 查看可以使用的cuda版本

  • 进入官网:https://developer.nvidia.com/cuda-toolkit-archive  找到并下载和显卡信息相匹配的cuda

  • 下载完成后按照默认的执行下去(当然也可以修改安装的路径),这里建议取消  VS Integration

  • 然后一步步往下执行

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

  • cuda的环境变量在安装时自动配置了,所以不需要我们操心。如果没有配置好,需要手动添加

  • 下面安装对应版本的cudnn:  https://developer.nvidia.com/cudnn  在这个网站先创建账户,再下载对应版本的cudnn

  • cudnn下载完成后进行解压,里面有三个文件夹

  • 将上面的三个文件夹覆盖你之前安装的cuda路径下的相同名称的文件夹(默认安装的cuda和我截图中的路径一样)

  • 下面我们使用anaconda安装tensorflow-gpu,【需要特别注意的是:需要确认TF、Python、cuDNN、CUDA,各个版本之间的关系,关系不对应安装过程无警告,但是GPU无法使用。对应的关系可以在这里查到:https://www.tensorflow.org/install/source_windows#gpu

  •  测试结果:

 

 

 

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值