yolov5运行遇到的问题汇总

错误提示： AssertionError: Invalid CUDA ‘–device 1‘ requested, use ‘–device cpu‘ or pass valid CUDA device(s)
解决 ：在跑yolov5时，cuda安装正确，print(torch.cuda.is_available())输出的是True，但一直报错，可能是我训练是用的是 --device=’ '，没有检测到显卡，后来我把 train.py 中的 default=‘cuda:0’，更改这个默认值就可以了
命令行输入 --device cuda依旧会报这个错误，删掉就好了。