模型应用
海光onnxruntime,使用rocm跑目前没问题,migraphx后端在添加的时候奔溃了,请问这是什么原因?
2025-11-20 10:05:49
318
![attachmentId-db009430771af73506ff38f7b28f9289.png](/api-forum/file/Redirect?path=forum/zzh/1763604238656/db009430771af73506ff38f7b28f9289.png) 我用image.sourcefind.cn:5000/dcu/admin/base/migraphx:5.1.0-ubuntu22.04-dtk25.04.2-py3.10 这个镜像 ![attachmentId-f0d40d1a7b6eab76a44483eb515b5474.png](/api-forum/file/Redirect?path=forum/zzh/1763604252718/f0d40d1a7b6eab76a44483eb515b5474.png)
DCU达人计划
全部评论(4)
只看楼主
M
minfy
1楼
您好,把代码里其中一行改为OrtMIGraphXProviderOptions migraphx_options = {};,确保OrtMIGraphXProviderOptions结构体成员清零初始化。
2025/12/11
S
suncd
回复 @minfy
我用python也是导致容器崩溃,如何解决 # 将GPU数据转换为MIGraphX参数
        parameter_map[input_name] = migraphx.gpudata_to_argument(
            shape=migraphx.shape(type="float_type", lens=[1, 3, 640, 640]),
            address=input_tensor.contiguous().data_ptr()
        )
        
        # 执行推理
        start_time = time.time()
        outputs = self.model.run(parameter_map) 执行到这里就崩了
1/16
M
minfy
回复 @suncd
您好,请提供一下DCU型号,镜像和报错截图,目前的报错和这个帖子是同一个测试环境吗?
1/16
S
suncd
回复 @minfy
Z100,镜像:jupyterlab-migraphx:4.3.0-ubuntu20.04-dtk24.04.1-py3.10,没有报错截图,容器直接就退出了
1/16
没有更多内容了
--
楼主
Level.2
0 帖子
楼主热帖
暂无数据