Drive AGX Orin upgraded to 6.0.10, but no cu pytorch

Autonomous Vehicles DRIVE AGX Orin DRIVE AGX Orin General
1

Please provide the following info (tick the boxes after creating this topic):
Software Version
[√] DRIVE OS 6.0.10.0
DRIVE OS 6.0.8.1
DRIVE OS 6.0.6
DRIVE OS 6.0.5
DRIVE OS 6.0.4 (rev. 1)
DRIVE OS 6.0.4 SDK
other

Target Operating System
[√] Linux
QNX
other

Hardware Platform
DRIVE AGX Orin Developer Kit (940-63710-0010-300)
DRIVE AGX Orin Developer Kit (940-63710-0010-200)
[√] DRIVE AGX Orin Developer Kit (940-63710-0010-100)
DRIVE AGX Orin Developer Kit (940-63710-0010-D00)
DRIVE AGX Orin Developer Kit (940-63710-0010-C00)
DRIVE AGX Orin Developer Kit (not sure its number)
other

SDK Manager Version
2.1.0
[√] other

Host Machine Version
native Ubuntu Linux 20.04 Host installed with SDK Manager
[√] native Ubuntu Linux 20.04 Host installed with DRIVE OS Docker Containers
native Ubuntu Linux 18.04 Host installed with DRIVE OS Docker Containers
other

Issue Description
My Drive AGX Orin development board was upgraded to 6.0.10, but I found that there is no cu version of pytorch. There is no matching version on the pytorch official website. How should I install the cu version of pytorch?

Thank you.

2

Dear @AlanWang,
We don’t officially support PyTorch on DRIVE. We recommend to develop your PyTorch model on host and use TRT to optimize your DNN model and perform inference.
Please see PyTorch + CUDA11.4 on 6.0.8.1 helps?

4

Thank you for reply.

6

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