RuntimeError: CUDA out of memory. Tried to allocate

python train.py --img 640 --batch 16 --epochs 5 --data coco128.yaml --weights yolov5s.pt

把batchsize改小

python train.py --img 640 --batch 2 --epochs 5 --data coco128.yaml --weights yolov5s.pt

能跑

(base) C:\Users\Administrator>activate yolo

(yolo) C:\Users\Administrator>cd I:\01ldzx\YOLO\yolov5\yolov5

(yolo) C:\Users\Administrator>I:

(yolo) I:\01ldzx\YOLO\yolov5\yolov5>python train.py --img 640 --batch 2 --epochs
 5 --data coco128.yaml --weights yolov5s.pt
Using torch 1.7.1 CUDA:0 (GeForce GTX 950, 2048MB)

Namespace(adam=False, batch_size=2, bucket='', cache_images=False, cfg='', data=
'coco128.yaml', device='', epochs=5, evolve=False, exist_ok=False, global_rank=-
1, hyp='data/hyp.scratch.yaml', image_weights=False, img_size=[640, 640], local_
rank=-1, log_artifacts=False, log_imgs=16, multi_scale=False, name='exp', noauto
anchor=False, nosave=False, notest=False, project='runs/train', rect=False, resu
me=False, save_dir='runs\\train\\exp12', single_cls=False, sync_bn=False, total_
batch_size=2, weights='yolov5s.pt', workers=1, world_size=1)
Start Tensorboard with "tensorboard --logdir runs/train", view at http://localho
st:6006/
Hyperparameters {'lr0': 0.01, 'lrf': 0.2, 'momentum': 0.937, 'weight_decay': 0.0
005, 'warmup_epochs': 3.0, 'warmup_momentum': 0.8, 'warmup_bias_lr': 0.1, 'box':
 0.05, 'cls': 0.5, 'cls_pw': 1.0, 'obj': 1.0, 'obj_pw': 1.0, 'iou_t': 0.2, 'anch
or_t': 4.0, 'fl_gamma': 0.0, 'hsv_h': 0.015, 'hsv_s': 0.7, 'hsv_v': 0.4, 'degree
s': 0.0, 'translate': 0.1, 'scale': 0.5, 'shear': 0.0, 'perspective': 0.0, 'flip
ud': 0.0, 'fliplr': 0.5, 'mosaic': 1.0, 'mixup': 0.0}

                 from  n    params  module                                  argu
ments
  0                -1  1      3520  models.common.Focus                     [3,
32, 3]
  1                -1  1     18560  models.common.Conv                      [32,
 64, 3, 2]
  2                -1  1     19904  models.common.BottleneckCSP             [64,
 64, 1]
  3                -1  1     73984  models.common.Conv                      [64,
 128, 3, 2]
  4                -1  1    161152  models.common.BottleneckCSP             [128
, 128, 3]
  5                -1  1    295424  models.common.Conv                      [128
, 256, 3, 2]
  6                -1  1    641792  models.common.BottleneckCSP             [256
, 256, 3]
  7                -1  1   1180672  models.common.Conv                      [256
, 512, 3, 2]
  8                -1  1    656896  models.common.SPP                       [512
, 512, [5, 9, 13]]
  9                -1  1   1248768  models.common.BottleneckCSP             [512
, 512, 1, False]
 10                -1  1    131584  models.common.Conv                      [512
, 256, 1, 1]
 11                -1  1         0  torch.nn.modules.upsampling.Upsample    [Non
e, 2, 'nearest']
 12           [-1, 6]  1         0  models.common.Concat                    [1]

 13                -1  1    378624  models.common.BottleneckCSP             [512
, 256, 1, False]
 14                -1  1     33024  models.common.Conv                      [256
, 128, 1, 1]
 15                -1  1         0  torch.nn.modules.upsampling.Upsample    [Non
e, 2, 'nearest']
 16           [-1, 4]  1         0  models.common.Concat                    [1]

 17                -1  1     95104  models.common.BottleneckCSP             [256
, 128, 1, False]
 18                -1  1    147712  models.common.Conv                      [128
, 128, 3, 2]
 19          [-1, 14]  1         0  models.common.Concat                    [1]

 20                -1  1    313088  models.common.BottleneckCSP             [256
, 256, 1, False]
 21                -1  1    590336  models.common.Conv                      [256
, 256, 3, 2]
 22          [-1, 10]  1         0  models.common.Concat                    [1]

 23                -1  1   1248768  models.common.BottleneckCSP             [512
, 512, 1, False]
 24      [17, 20, 23]  1    229245  models.yolo.Detect                      [80,
 [[10, 13, 16, 30, 33, 23], [30, 61, 62, 45, 59, 119], [116, 90, 156, 198, 373,
326]], [128, 256, 512]]
Model Summary: 283 layers, 7468157 parameters, 7468157 gradients, 17.5 GFLOPS

Transferred 370/370 items from yolov5s.pt
Optimizer groups: 62 .bias, 70 conv.weight, 59 other
Scanning '..\coco128\labels\train2017.cache' for images and labels... 128 found
Scanning '..\coco128\labels\train2017.cache' for images and labels... 128 found
Analyzing anchors... anchors/target = 4.26, Best Possible Recall (BPR) = 0.9946
Image sizes 640 train, 640 test
Using 1 dataloader workers
Logging results to runs\train\exp12
Starting training for 5 epochs...

     Epoch   gpu_mem       box       obj       cls     total   targets  img_size

Scanning '..\coco128\labels\train2017.cache' for images and labels... 128 found
       0/4    0.847G   0.04502   0.07511   0.02855    0.1487        19       64
               Class      Images     Targets           P           R      mAP@.
                 all         128         929       0.341       0.781        0.69
       0.455

     Epoch   gpu_mem       box       obj       cls     total   targets  img_size

       1/4    0.847G   0.04517   0.07459    0.0295    0.1493        40       64
               Class      Images     Targets           P           R      mAP@.
                 all         128         929       0.325       0.802       0.703
       0.447

     Epoch   gpu_mem       box       obj       cls     total   targets  img_size

       2/4    0.847G   0.04585   0.07038   0.02843    0.1447        18       64
               Class      Images     Targets           P           R      mAP@.
                 all         128         929       0.322       0.802       0.707
       0.461

     Epoch   gpu_mem       box       obj       cls     total   targets  img_size

       3/4    0.847G    0.0441   0.07255    0.0299    0.1466        14       64
               Class      Images     Targets           P           R      mAP@.
                 all         128         929       0.306       0.802       0.719
       0.462

     Epoch   gpu_mem       box       obj       cls     total   targets  img_size

       4/4    0.847G   0.04388   0.06547   0.02902    0.1384        22       64
               Class      Images     Targets           P           R      mAP@.
                 all         128         929       0.301       0.799       0.723
       0.477
Optimizer stripped from runs\train\exp12\weights\last.pt, 15.2MB
Optimizer stripped from runs\train\exp12\weights\best.pt, 15.2MB
5 epochs completed in 0.060 hours.