search

dianjixz / v831_yolo

GNU General Public License v2.0
15 stars 10 forks source link

RuntimeError: No CUDA GPUs are available #1

Open mask85 opened 2 years ago

mask85 commented 2 years ago

执行 python train.py -d custom --cuda -v slim_yolo_v2 -hr -ms RuntimeError: No CUDA GPUs are available

dianjixz commented 2 years ago

如果使用 cpu 运行时，运行这个命令 python train.py -d custom -v slim_yolo_v2 -hr -ms