Skip to content

yolov10在CUDA推理结果异常 #4613

Description

@zhiweiwo

-MNN 3.6.0

  • GPU:RTX 5090
  • 驱动 580.159.03,CUDA 12.8
  • 模型 FP32,无量化
  • 转换:MNNConvert -f ONNX --modelFile best.onnx --MNNModel best.mnn --bizCode MNN

尝试使用backendTest.out工具测试

export LD_LIBRARY_PATH=/path/to/MNN/build_cuda:/path/to/MNN/build_cuda/source/backend/cuda:/usr/local/cuda-12.8/lib64:$LD_LIBRARY_PATH
./backendTest.out best.mnn 2 0.05 0 0

输出:

Correct for 94, name=silu_28
0: -0.718346 != 0.670780
95 [max_pool2d] Op outputs max_pool2d is error

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions