yolov7的bug,无法指定显卡(程序默认0号卡)

**问题:**命令行参数指定的是4号卡,但实际却总是在0号卡建立进程
真抽象啊,这一步,模型被送到0号卡,但实际上,送到了4号卡(进程是在4号卡上建立的)yolov7的bug,无法指定显卡(程序默认0号卡)_第1张图片
解决办法:
在train.py的train()函数下,第一行选择显卡:

device=torch.device('cuda:4')

自己指定显卡即可,也可以在命令行添加一遍,不冲突,但要一样的

你可能感兴趣的:(yolov7,YOLO,bug,yolov7,无法指定显卡)