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 就好。
然后选择创建一个新的还是远来已经创建好的,就可以:

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

被折叠的 条评论
为什么被折叠?



