@rahulswa if you want everything to be on CUDA 11.8, you would need to rebuild PyTorch with CUDA 11.8 instead. Or build pytorch-3d with CUDA 11.4 instead. If you go with the former, you can find instructions on building PyTorch from source in this thread:
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Upgrading CUDA 11.4 to CUDA 11.8 | 3 | 1050 | September 9, 2024 | |
| Which Cuda and pytroch with Jetpack 6 on Jetson agx orin dev kit | 2 | 67 | November 14, 2024 | |
| Unable to install pytorch with cuda support on jetson AGX Xavier | 2 | 1071 | January 22, 2024 | |
| Jetson AGX Xavier Jetpack 5 upgraded to CUDA 11.8 can't use torch :( | 5 | 411 | May 13, 2024 | |
| Unable to use cuda cores for yolo. tried everything found on internet for orin nano dev kit | 6 | 71 | February 7, 2025 | |
| Using Pytorch on GPU after upgrading cuda to 11.8 on Jetpack5 | 8 | 1811 | August 7, 2023 | |
| Pytorch with CUDA support Jetson Orin Nano | 4 | 1242 | May 14, 2024 | |
| PyTorch for Jetson Orin Jetpack 5.1 installation issue | 12 | 1346 | January 23, 2024 | |
| Pytorch with Cuda 11.8 on Jetson ORIN AGX ( aarch64 ) | 5 | 1800 | February 13, 2024 | |
| Pytorch for jetson nano orin does not work | 5 | 1383 | August 22, 2023 |