上传你的onnx文件，我们将分析它来知道
GPU环境下如何使用importONNXfunction ?
3次浏览(过去30天)
显示旧的注释
我找不到使用importONNXfunction在gpu环境中使用它的方法。
这是代码:
parallel.gpu.enableCUDAForwardCompatibility(真正的)
I = gpuArray(I);
params = importONNXFunction(modelfile，'UNet177Fcn');
result = UNet184Fcn(I,params，'Training'，false);
当我改变输入到gpu数组，gpu工作，但结果出错。
有这么多占位符需要替换，所以我认为我的模型更适合importONNXFunction。
这就是模型的问题吗?或者只是importonnx函数不适合gpu??
答案(1)
Anshika Chaurasia
2021年6月16日
你好,
我尝试使用importONNXFunction导入一个onnx文件，它在传递gpuArray输入时给了我正确的结果。gpu环境下的importONNXFunction没有问题。可能您使用的型号有问题。
希望能有所帮助!
0评论
另请参阅
您也可以从以下列表中选择一个网站: