yolov11配置环境,实现OBB带方向目标检测

最新推荐文章于 2026-06-17 17:10:13 发布

原创

最新推荐文章于 2026-06-17 17:10:13 发布 · 3.1k 阅读

1. 安装GPU驱动，cuda

验证安装,nvidia-smi命令查看状态

nvcc -V查看cuda版本, 按安装后提示配置路径。

注意：

cuda选择版本，对pytorch支持有关系，目前pytorch最高支持到11.8 ，实测A5000装530驱动，安装cuda12.2会无法正确运行，可以降级安装cuda11.8，可以完美匹配。

2. 安装pytorch

用conda安装，确保版本兼容

conda install pytorch

验证安装正确，且cuda支持OK，

3. 安装python虚拟环境，方便测试

安装配置miniconda

miniconda安装-CSDN博客

4. yolo环境配置

仓库：

GitHub - ultralytics/ultralytics: Ultralytics YOLO11 🚀

官方文档，安装最好用conda方式：

快速入门 -Ultralytics YOLO 文档

安装过程遇到的错误及排查：

undefined symbol: __nvJitLinkComplete_12_4, version libnvJitLink.so.12
是torch版本匹配问题，降级到cuda11.8

官方有conda安装方式，但是外网限制问题，无法安装，会报错，连接不上conda

conda install -c conda-forge ultralytics

git安装

git目录后用pip安装可以正常使用

pip install -e .

测试python导入环境

命令行测试图片：

yolo predict model=yolo11s.pt source='https://ultralytics.com/images/bus.jpg' imgsz=320

命令行预测（predict）视频：

yolo predict model=yolo11n-seg.pt source=./1.mp4 imgsz=320

验证训练流程：

yolo train data=coco8.yaml model=yolo11n.pt epochs=10 lr0=0.01

训练结果


(py39yolov11) hy@hy-C621-WD12-IPMI:~/yolopan/ultralytics$ yolo train data=coco8.yaml model=yolo                                          11n.pt epochs=10 lr0=0.01
Ultralytics 8.3.63 🚀 Python-3.9.21 torch-2.5.1+cu124 CUDA:0 (NVIDIA RTX A5000, 24248MiB)
engine/trainer: task=detect, mode=train, model=yolo11n.pt, data=coco8.yaml, epochs=10, time=Non                                          e, patience=100, batch=16, imgsz=640, save=True, save_period=-1, cache=False, device=None, work                                          ers=8, project=None, name=train2, exist_ok=False, pretrained=True, optimizer=auto, verbose=True                                          , seed=0, deterministic=True, single_cls=False, rect=False, cos_lr=False, close_mosaic=10, resu                                          me=False, amp=True, fraction=1.0, profile=False, freeze=None, multi_scale=False, overlap_mask=T                                          rue, mask_ratio=4, dropout=0.0, val=True, split=val, save_json=False, save_hybrid=False, conf=N                                          one, iou=0.7, max_det=300, half=False, dnn=False, plots=True, source=None, vid_stride=1, stream                                          _buffer=False, visualize=False, augment=False, agnostic_nms=False, classes=None, retina_masks=F                                          alse, embed=None, show=False, save_frames=False, save_txt=False, save_conf=False, save_crop=Fal                                          se, show_labels=True, show_conf=True, show_boxes=True, line_width=None, format=torchscript, ker                                          as=False, optimize=False, int8=False, dynamic=False, simplify=True, opset=None, workspace=None,                                           nms=False, lr0=0.01, lrf=0.01, momentum=0.937, weight_decay=0.0005, warmup_epochs=3.0, warmup_                                          momentum=0.8, warmup_bias_lr=0.1, box=7.5, cls=0.5, dfl=1.5, pose=12.0, kobj=1.0, nbs=64, hsv_h                                          =0.015, hsv_s=0.7, hsv_v=0.4, degrees=0.0, translate=0.1, scale=0.5, shear=0.0, perspective=0.0                                          , flipud=0.0, fliplr=0.5, bgr=0.0, mosaic=1.0, mixup=0.0, copy_paste=0.0, copy_paste_mode=flip,                                           auto_augment=randaugment, erasing=0.4, crop_fraction=1.0, cfg=None, tracker=botsort.yaml, save                                          _dir=/home/hy/yolopan/ultralytics/runs/detect/train2

                   from  n    params  module                                       arguments                                             
  0                  -1  1       464  ultralytics.nn.modules.conv.Conv             [3, 16, 3, 2                                          ]
  1                  -1  1      4672  ultralytics.nn.modules.conv.Conv             [16, 32, 3,                                           2]
  2                  -1  1      6640  ultralytics.nn.modules.block.C3k2            [32, 64, 1,                                           False, 0.25]
  3                  -1  1     36992  ultralytics.nn.modules.conv.Conv             [64, 64, 3,                                           2]
  4                  -1  1     26080  ultralytics.nn.modules.block.C3k2            [64, 128, 1,                                           False, 0.25]
  5                  -1  1    147712  ultralytics.nn.modules.conv.Conv             [128, 128, 3                                          , 2]
  6                  -1  1     87040  ultralytics.nn.modules.block.C3k2            [128, 128, 1                                          , True]
  7                  -1  1    295424  ultralytics.nn.modules.conv.Conv             [128, 256, 3                                          , 2]
  8                  -1  1    346112  ultralytics.nn.modules.block.C3k2            [256, 256, 1                                          , True]
  9                  -1  1    164608  ultralytics.nn.modules.block.SPPF            [256, 256, 5                                          ]
 10                  -1  1    249728  ultralytics.nn.modules.block.C2PSA           [256, 256, 1                                          ]
 11                  -1  1         0  torch.nn.modules.upsampling.Upsample         [None, 2, 'n                                          earest']
 12             [-1, 6]  1         0  ultralytics.nn.modules.conv.Concat           [1]                                                   
 13                  -1  1    111296  ultralytics.nn.modules.block.C3k2            [384, 128, 1                                          , False]
 14                  -1  1         0  torch.nn.modules.upsampling.Upsample         [None, 2, 'n                                          earest']
 15             [-1, 4]  1         0  ultralytics.nn.modules.conv.Concat           [1]                                                   
 16                  -1  1     32096  ultralytics.nn.modules.block.C3k2            [256, 64, 1,                                           False]
 17                  -1  1     36992  ultralytics.nn.modules.conv.Conv             [64, 64, 3,                                           2]
 18            [-1, 13]  1         0  ultralytics.nn.modules.conv.Concat           [1]                                                   
 19                  -1  1     86720  ultralytics.nn.modules.block.C3k2            [192, 128, 1                                          , False]
 20                  -1  1    147712  ultralytics.nn.modules.conv.Conv             [128, 128, 3                                          , 2]
 21            [-1, 10]  1         0  ultralytics.nn.modules.conv.Concat           [1]                                                   
 22                  -1  1    378880  ultralytics.nn.modules.block.C3k2            [384, 256, 1                                          , True]
 23        [16, 19, 22]  1    464912  ultralytics.nn.modules.head.Detect           [80, [64, 12                                          8, 256]]
YOLO11n summary: 319 layers, 2,624,080 parameters, 2,624,064 gradients, 6.6 GFLOPs

Transferred 499/499 items from pretrained weights
Freezing layer 'model.23.dfl.conv.weight'
AMP: running Automatic Mixed Precision (AMP) checks...
AMP: checks passed ✅
train: Scanning /home/hy/yolopan/datasets/coco8/labels/train.cache... 4 images, 0 backgrounds,
val: Scanning /home/hy/yolopan/datase

标签