Ubuntu 16.04+tensorflow-gpu运行yolov3算法

本文分享了作者在Ubuntu 16.04环境下,如何成功安装GTX1650显卡驱动、配置TensorFlow-gpu 1.11.0环境,以及使用GitHub上的YunYang1994/tensorflow-yolov3项目进行深度学习实践的过程,包括了遇到的问题和解决方案。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

导读:最近有一个有关于机器人与深度学习结合的项目,我就开始研究了yolo系列的算法,主流的 框架是tensorflow、pytorch、caffe、darknet等等,于是尝试了GitHub上的代码https://github.com/YunYang1994/tensorflow-yolov3,效果还不错!

本人电脑的环境

显卡:GTX1650(4G独显)+i7(16G主显)+256G SSD+1T机械硬盘

驱动:470.63.01 

cuda:9.0+cudnn:7.4.2+tensorflow-gpu 1.11.0版本

 一、安装过程

1.之前新买的戴尔电脑,装好了ROS系统,5月初刚好买了一个显示屏,办公双显很爽,但Ubuntu16.04系统需要安装显卡驱动才能双显,莫名其妙按照网上的装显卡的方法,尝试了10多遍,前面的步骤一直都正常,到了最后一步还是无法解压,搜遍了全网,还是无法解决,咨询了一些人,好像是内核版本低引起的;当时我也配置好ROS的开发环境了,升级内核有可能引起系统崩溃,要面临重装系统,避免这么麻烦,于是放弃,先用段时间看看;直到9月初开始要用深度学习搭建环境,之前的ROS开发也差不多成熟了,于是铁定决心重装系统,配置好新的环境。

Ubuntu16.04下安装显卡驱动记录 - lllittletree - 博客园

2.重装系统后,为了避免环境的混乱,我一开始是先装好深度学习环境,tensorflow,darknet,后面才装ROS环境,心得:

在安装搭建深度学习环境肯定遇到不少问题,在搭建深度学习环境之前,先大概把整个流程了解清楚,比如torch tf keras 等都在虚拟环境,caffe和DarkNet只能在主环境跑,用anaconda搭建虚拟环境,以免相互影响。遇到具体的问题,及时具体分析,并尝试多种方法,这一篇博客记录一下安装过程,以便后面查阅,供大家参考一下。

3.因为GitHub上的项目用的环境是tensorflow-gpu 1.11.0版本,然后结合这个表格,于是对应的安装好cuda和cudnn版本,安装过程是参考了这篇博客https://blog.youkuaiyun.com/zywvvd/article/details/86355060

4.安装anaconda3一些坑 

不要直接安装Anaconda3-2018.12-Linux-x86_64.sh,因为这个包会自带安装python3.7,但是根据上面的表格,tensorflow-gpu版本不支持3.7,如果直接安装,要自己手动从py3.7降级到py3.6,太多坑了,搞不好又要重装conda环境

于是下载安装了Anaconda3-5.2.0-Linux-x86_64.sh,然后我再直接在conda创建的虚拟环境下,手动装上py3.6版本,完美解决

5.如果中间遇到缺乏一些什么依赖和库,搜索一下问题,安装对应的包就可以解决

二、测试代码

​​​​​​GitHub - YunYang1994/tensorflow-yolov3: 🔥 TensorFlow Code for technical report: "YOLOv3: An Incremental Improvement"

按照作者的步骤一步一步执行,中间可能缺少一些包,当时没来得及把问题截图记录下来,但是问题都不大,最后直接测试图片和视频,效果不错

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值