Upgrading cuda

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
[1]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
[1]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)
[1]DRIVE AGX Orin Developer Kit (not sure its number)
other

SDK Manager Version
2.1.0
[1] 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
Hi team, I have cuda 11.4 installed on my system, but in order to use pytorch with cuda support, I need to have latest version of cuda, How can I upgrade my cuda version?
please help!

5

Dear @akshay.tupkar ,
Upgrading of CUDA alone is not possible as it is part of DRIVE OS package.

6

Dear @SivaRamaKrishnaNV
Okay, I got it.
Does nvidia provide any pytorch wheels for cuda 11.4?
If I get this, it would be really helpful

7

Dear @akshay.tupkar ,
I notice a topic for requesting PyTorch at
Please check Installation of pytorch with cuda support on drive orin - #5 by SivaRamaKrishnaNV. Please see if it helps?

8

Dear @SivaRamaKrishnaNV

I get it but in release notes, I found out that there is a pytorch container which is pre built for cuda 11.4? The version of container is 21.07

So if I download the pytorch image and run it, will it work on drive orin?

9

yes it will on DRIVE. i reckon there is backward compatibility for cuda minor versions. you can install pre-built wheel or install from source for your cuda version, i would suggest building from source if you are planning on using opencv/open3d libs later as well.

1 Like
10

Dear @akshay.tupkar ,
I notice PyTorch and TF versions mentioned in software version table in release notes. it is related to TRT support on tegra in general. There are no seperate whl files or container for DRIVE.
I notice the whl files from the Jetson are working. Please give a try and file a new topic incase you have any issue.

13

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