错误提示: 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 首次发布
本文介绍了解决在使用YoloV5时遇到的CUDA设备请求无效问题的方法。当训练模型时指定的CUDA设备不正确或者未被正确识别时,可以通过修改训练脚本中的默认设备设置来解决这一问题。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
2503





