How to use importONNXfunction in GPU environment?

2 次查看（过去 30 天）
SeungHyun Jang
SeungHyun Jang 2021-6-14
评论： cui,xingxing 2021-8-3
I can't find the way to use importONNXfunction to use it at the gpu enviroment.
This is the code:
parallel.gpu.enableCUDAForwardCompatibility(true)
I = gpuArray(I);
params = importONNXFunction(modelfile,'UNet177Fcn');
result = UNet184Fcn(I,params,'Training',false);
when I change the input to the gpuarray, gpu works but the results going wrong.
there are so many placeholders to be replaced, so I think my model is more fit with the importONNXFunction.
So, is this the problem about the model? or just the importONNXfunction is doesn't fit with the gpu??

请先登录，再回答此问题。

回答（1 个）

Anshika Chaurasia
Anshika Chaurasia 2021-6-16
Hi,
I tried to use importONNXFunction to import a onnx file and it gave me correct results on passing gpuArray input. There is no issue of importONNXFunction with gpu environment. May be there is some issue in the model you are using.
Hope it helps!

类别

在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息

标签

产品


版本

R2020b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by