HZAI-ZJNU / Mamba-YOLO

the official pytorch implementation of “Mamba-YOLO:SSMs-based for Object Detection”
Apache License 2.0
208 stars 24 forks source link

RuntimeError: No CUDA GPUs are available #27

Open xsa12345 opened 1 month ago

xsa12345 commented 1 month ago

2024-07-17 09-39-49 的屏幕截图

yaosx425 commented 1 month ago

2024-07-17 09-39-49 的屏幕截图

没有gpu可用,只能使用cpu

EthanW-coder commented 1 month ago

Hello, this issue comes from the fact that you are using the default parameters for multi-GPU training in mbyolo_train.py. If you are training with only one GPU, could you please change parser.add_argument('--device', default='0,1,2,3,4,5,6,7', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') to parser.add_argument('--device', default='0', help='cuda device, i.e. 0 or 0,1,2,3 or cpu') to parser.add_argument('--device', default='0', help='cuda device'). argument('--device', default='0', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')