when training YOLOv5, GPU utilization I want to use fully GPU So i checked and correct "number of workers",anything i can do but my GPU did not work fully please help me before crash my computer
Asked
Active
Viewed 174 times
1 Answers
0
Try to move your network (I call it yolo_network) to CUDA
yolo_network.to('cuda')
blue_lama
- 125
- 7
-
very very very very useless – hashiburae Mar 25 '23 at 11:35