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