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 DRIV…

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

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

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?

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.

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 ne…

Upgrading cuda

Autonomous Vehicles DRIVE AGX Orin DRIVE AGX Orin General

SivaRamaKrishnaNV September 29, 2025, 12:55pm 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?

Topic		Replies	Views
Installation of pytorch with cuda support on drive orin DRIVE AGX Orin General pytorch , driveos-dl	2	138	October 27, 2025
Update CUDA version in DRIVE OS DRIVE AGX Orin General driveos-cuda	2	174	August 6, 2024
Update CUDA version in DRIVE OS DRIVE AGX Orin General driveos-cuda	2	137	July 18, 2024
Drive AGX Orin upgraded to 6.0.10, but no cu pytorch DRIVE AGX Orin General driveos-dl	3	135	September 22, 2024
Independently update CUDA version in DRIVE OS DRIVE AGX Orin General driveos-cuda	6	788	May 18, 2023
Upgrading CUDA 11.4 to CUDA 11.8 Jetson Orin NX cuda	3	1819	September 9, 2024
Upgrade to latest CUDA on Jetson without upgrading JetPack Jetson AGX Orin	3	4821	September 29, 2023
Different cuda versions installed and cuda unavailable \| Jetson Orin NX Jetson AGX Orin cuda , pytorch , cudnn	7	1450	November 20, 2023
How to use pytorch 2.3 + torchaudio in driver os DRIVE AGX Orin General driveos-dl	3	201	June 24, 2025
PyTorch + CUDA11.4 on 6.0.8.1 DRIVE AGX Orin General pytorch , driveos-cuda	11	2595	February 7, 2024

Upgrading cuda

Related topics