Tensorflow object detection API 搭建自带案例的物体识别模型 环境搭建及测试

一、环境搭建
1、安装pycharm3.6.4及对应版本的Anaconda
安装包百度云链接:
百度云链接 提取码:m7xa

2、安装Tensorflow
Tensorflow有CPU版本与GPU版本
通过pip命令安装CPU版本: pip install tensorflow
通过pip命令安装GPU版本: pip install tensorflow-gpu
如果速度太慢,可使用清华镜像:

pip install (包名称) -i https://pypi.tuna.tsinghua.edu.cn/simple

测试tensorflow:
打开命令行,输入python进入python环境
在输入

import tensorflow as tf

不报错即可,如果出现警告,可能是因为部分包的版本太高的问题

测试tensorflow-gpu:
打开命令行,输入python进入python环境

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

出现类似输出结果即为安装成功

Sample Output
 
[name: "/cpu:0" device_type: "CPU" memory_limit: 268435456 locality { } incarnation: 4402277519343584096,
name: "/gpu:0" device_type: "GPU" memory_limit: 6772842168 locality { bus_id: 1 } incarnation: 7471795903849088328 physical_device_desc: "device: 0, name: GeForce GTX 1070, pci bus id: 0000:05:00.0" ]

3、安装CUDA、cuDNN
对于tensorflow-gpu还需另外安装CUDA、cuDNN
本人的tensorflow-gpu版本: 1.15.0
tensorflow版本:1.14.0
CUDA版本: 10.0
各位安装的时候要注意版本的匹配问题!!!

CUDA安装包百度云链接(是一个exe文件,双击安装):

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值