Search before asking

• I have searched the YOLOv5 issues and found no similar bug report.

YOLOv5 Component

Training, Multi-GPU

Bug

All GPUs are utilized by torch 1.12 with current YOLOv5 master when a single-GPU command is run, i.e.:

Error does not occur with torch==1.11

@AyushExel FYI

Environment

Docker image

Minimal Reproducible Example

python train.py --epochs 10 --device 7

Additional

Temp workaround is to use torch 1.11

pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!