自动驾驶-YOLOV5网络训练(可能会遇到五个坑)

本文档详细记录了在Ubuntu20.04环境下使用Pytorch1.10.2进行YOLOV5自动驾驶网络训练时可能遇到的问题,包括环境配置、wandb api_key设置、CUDA内存不足、模型依赖冲突以及训练输入尺寸错误等,并提供了相应的解决办法和参考链接。

1、环境:

操作系统:Ubuntu20.04

Python3.9

Pytorch1.10.2

IDE:(本人使用的是ECLIPSE)

显卡:NVIDIA GeForce GTX 1060 6GB

2、网络训练

在YOLOV5目标下,选择:train.py,执行

3、可能会遇到的坑一

报:/bin/sh: pip: command not found

遇到这错误,说明环境不协调,需要安装配套的环境,具体每个人的环境不一样,需要细心解决,找资料。

4、可能会遇到的坑二

报:wandb.errors.UsageError: api_key not configured (no-tty). call wandb.login(key=[your_api_key])

这是需要在终端运行:

wandb init

然后,会有一个提示网址,打开网页进入

 

注册登录后(如果有帐号,则直接登录即可),会生成一个KEY

 


在终端中输入(ctrl + v  在终端中不会出现你粘贴的东西),粘贴后 enter 就好。

然后选择创建一个新的还是远来已经创建好的,就可以:

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值