How to install pytorch on px2?

Autonomous Vehicles DRIVE PX 2 General
1

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

2

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.

3

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

4

Hi, are there any updates if torch will be available on the drive px2?

5

No, sorry for that.

6

Hi,

here is the compilation/installation guide for PyTorch (1.7.1) on Drive PX2. Hope it helps!!

Cheers,
Vishal Balaji

7

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.