技术栈
无法指定显卡
毕竟是shy哥
1 年前
yolo
·
yolov8
·
yolov7
·
无法指定显卡
解决yolo无法指定显卡的问题,实测v5、v7、v8有效
基本上这个就能解决了!!! 在train.py的最上方加上下面这两行,注意是最上面,其次指定的就是你要使用的显卡