Win10+GTX1650显卡下安装Tensorflow-gpu1.14的踩坑过程及训练目标检测模型

Win10+GTX1650显卡下安装Tensorflow-gpu1.14的踩坑过程及训练目标检测模型

作为一个刚接触深度学习的小白,因目标检测的任务需求,在网上查阅了大量前辈写的相关blog,学到了很多东西,最后自己搭建完环境训练了模型。因为搭配环境时踩了很多坑,我将内容整理一下,方便必要时再次查看。

安装tensorflow-gpu前python、cuda、cudnn、vs的版本需要一一对应,否则可能最后会报错。我装的比较新:
anaconda3
python3.7
cuda10.1
cudnn7.6
vs2019
tensorflow-gpu 1.14

安装tensorflow object detection API

1.去anaconda官网,下载完后自带python3.73
在这里插入图片描述

2.tensorflow分cpu版和gpu版,cup版相当容易,命令行pip install tensorflow即可,安装的是tensorflow1.14,gpu版相当麻烦,先跳过。

3.安装tensorflow object detection API,先去Github上下载https://github.com/tensorflow/models

然后安装配置Protobuf,下载地址https://github.com/protocolbuffers/protobuf/releases

具体的操作参考TensorFlow学习(四)之基于win10实现官方Object Detection API 步骤及出现的坑bug解决办法博主已经写的非常详细了,感谢博主,让我受益匪浅。

4.然后测试一下官方demo,在research\object_detection文件夹下打开jupyter notebook,因为我的版本是tensorflow1.14,我会把下面的注释掉或者4前面加个1。

  • 23
    点赞
  • 151
    收藏
    觉得还不错? 一键收藏
  • 25
    评论
评论 25
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值