Other thing you will need to grab a pre-built pytorch with magma support to have access to this or you will need to build your own pytorch with this resource enabled.
I don’t use anaconda so I cannot tell you for sure if there is pre-built package that recognizes or not if you have magma installed or not.