NVIDIA / NeMo

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)
https://docs.nvidia.com/nemo-framework/user-guide/latest/overview.html
Apache License 2.0
11.84k stars 2.46k forks source link

RuntimeError: fft: ATen not compiled with MKL support [Question] #1938

Closed ulucsahin closed 3 years ago

ulucsahin commented 3 years ago

Describe your question I am getting the error on title on arm processor when trying to do inference with QuartzNet model. This is expected since MKL is not supported on arm. How can do inference on arm devices? Are there any alternatives?

I am using PyTorch 1.9 but I think this is not the reason for this error.

Environment overview (please complete the following information)

Environment details

If NVIDIA docker image is used you don't need to specify these. Otherwise, please provide:

okuchaiev commented 3 years ago

At the moment we do not test/support Arm architectures. Do you know which module need MKL? Are you able to install Pytorch on Arm without MKL?