无法指定显卡

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