错误提示: 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
依旧会报这个错误,删掉就好了。
yolov5运行遇到的问题汇总
于 2022-12-06 23:05:07 首次发布