gpt4 book ai didi

pytorch - 如何在基于 arm 的 CPU 上模拟/运行使用 ATen STFT 实现的 PyTorch 模型?

转载 作者:行者123 更新时间:2023-12-04 07:56:07 39 4
gpt4 key购买 nike

我正在尝试在没有 gpu 的基于 arm 的设备上为 ASR 运行我的 PyTorch 模型。据我所知,arm 不支持 ATen 使用的 MKL。自然地,当我尝试进行推理时会收到以下错误:

RuntimeError: fft: ATen not compiled with MKL support
我怎么解决这个问题?我可以使用任何替代方法吗?

最佳答案

如果您的目标设备是移动设备,那么首先尝试使用 Pytorch Mobile 将其转换为 TorchScript 是合理的。 TorchScript 是可以在移动环境中运行的 PyTorch 模型的中间表示。
https://pytorch.org/mobile/home/

关于pytorch - 如何在基于 arm 的 CPU 上模拟/运行使用 ATen STFT 实现的 PyTorch 模型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66693151/

39 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com