No module named 'migraphx'

[liangyong928]

Description of errors

I have successfully built torch_migraphx, but when I try to import it, it always fails with an error " No module named 'migraphx' " I'm not sure why this is happening.

Attach any links, screenshots, or additional evidence you think will be helpful.

No response

[onesnep]

I had the same error. Setting an environmental variable to the rocm path like so solved it:

export PYTHONPATH=/opt/rocm/lib:$PYTHONPATH

The rocm directory is usually in /opt, though some distros place it elsewhere.

[FeepingCreature]

If you're on a new Ubuntu, you might be on Python 3.11. Migraphx (not torch_migraphx) only ships libs for Python 3.8 and 3.10. Check python --version.