Hello All,
I would like enable to cuda for torch for the object detection module. I have tried with multiple version for torch 2.3, 2.4 till 2.5 but no luck. Could anyone help me here to install this torch installation with cuda enabled on the Jetpack 6.1.
aravinthrajganesan@aravinthrajganesan-desktop:~$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Wed_Aug_14_10:14:07_PDT_2024
Cuda compilation tools, release 12.6, V12.6.68
Build cuda_12.6.r12.6/compiler.34714021_0
aravinthrajganesan@aravinthrajganesan-desktop:~$ python
Python 3.10.12 (main, Nov 6 2024, 20:22:13) [GCC 11.4.0] on linux
Type “help”, “copyright”, “credits” or “license” for more information.
import torch
print(torch.version)
2.5.0a0+872d972e41.nv24.08
print(“CUDA available:”, torch.cuda.is_available())
CUDA available: False
aravinthrajganesan@aravinthrajganesan-desktop:~$ cat /etc/nv_tegra_release
R36 (release), REVISION: 4.0, GCID: 37537400, BOARD: generic, EABI: aarch64, DATE: Fri Sep 13 04:36:44 UTC 2024
KERNEL_VARIANT: oot
TARGET_USERSPACE_LIB_DIR=nvidia
TARGET_USERSPACE_LIB_DIR_PATH=usr/lib/aarch64-linux-gnu/nvidia
aravinthrajganesan@aravinthrajganesan-desktop:~$