how to install pytorch on px2?

hi

i really want to have a test by using pytorch, and how to compiling it and how to set the cmake file is my problem.

thanks all

Hi,

I am having similar issues, wanting to test my pytorch model on px2. Didn’t find much comprehensive documents on installing it except on Jetson platforms. Will the steps be something similar?

Thank you.

You may try to build it from the source with the steps shared in the topic.
https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/
Even that’s for supporting on Jetson Nano, TX2 and Xavier, but you may get it a try if can work.