PyTorch won't detect CUDA 10.2 on Ubuntu 18.04 Jetpack

deployment
1

Currently working with a NVIDIA Jetson. nvcc outputs,

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Feb_28_22:34:44_PST_2021
Cuda compilation tools, release 10.2, V10.2.300
Build cuda_10.2_r440.TC440_70.29663091_0

I tried setting up PyTorch from this page with the following settings,

image
image825×337 18.2 KB

Despite this, whenever I run torch.cuda.is_available() in python, I get the “False” output. Are there any dependencies or compatibility issues that I may be unaware of? Thanks.

2

This post has some good suggestions My jetson nano board returns 'False' to torch.cuda.is_available() in local directory - Jetson Nano - NVIDIA Developer Forums