"No ML Engine" for Torch models

Hello,

I’m trying to run some ML models in Labels with GPU, but the option to select the torch models is missing. The message next to the models says “No ML Engine”.

In the venv that ImFusion Labels uses I have torch with cuda installed and have tested that cuda enabled models run in that venv. Are there additional requirements needed for ImFusion Labels to detect the presence of Torch?


Hi idlv,
our Linux Installer does not normally ship with Torch, but it does ship with Onnx. Can you use that instead? Those models also run on the GPU using cuda.