Hi @naikrohanp97, which version of JetPack-L4T are you running (you can check this with cat /etc/nv_tegra_release) and which PyTorch wheel did you install? You would need to install a PyTorch wheel from here which is compatible with your version of JetPack.
Also, can you check that you have CUDA Toolkit installed?
ls /usr/local/cuda/lib64/
libcublasLt.so libcufft_static.a libcurand_static.a libnppial.so.10 libnppif.so.10 libnppisu.so.10 libnvgraph.so.10.2.300
libcublasLt.so.10 libcufft_static_nocallback.a libcusolver.so libnppial.so.10.2.1.300 libnppif.so.10.2.1.300 libnppisu.so.10.2.1.300 libnvgraph_static.a
libcublasLt.so.10.2.3.300 libcufftw.so libcusolver.so.10 libnppial_static.a libnppif_static.a libnppisu_static.a libnvperf_host.so
libcublasLt_static.a libcufftw.so.10 libcusolver.so.10.3.0.300 libnppicc.so libnppig.so libnppitc.so libnvperf_target.so
libcublas.so libcufftw.so.10.1.2.300 libcusolver_static.a libnppicc.so.10 libnppig.so.10 libnppitc.so.10 libnvrtc-builtins.so
libcublas.so.10 libcufftw_static.a libcusparse.so libnppicc.so.10.2.1.300 libnppig.so.10.2.1.300 libnppitc.so.10.2.1.300 libnvrtc-builtins.so.10.2
libcublas.so.10.2.3.300 libcuinj64.so libcusparse.so.10 libnppicc_static.a libnppig_static.a libnppitc_static.a libnvrtc-builtins.so.10.2.300
libcublas_static.a libcuinj64.so.10.2 libcusparse.so.10.3.1.300 libnppicom.so libnppim.so libnpps.so libnvrtc.so
libcudadevrt.a libcuinj64.so.10.2.300 libcusparse_static.a libnppicom.so.10 libnppim.so.10 libnpps.so.10 libnvrtc.so.10.2
libcudart.so libculibos.a liblapack_static.a libnppicom.so.10.2.1.300 libnppim.so.10.2.1.300 libnpps.so.10.2.1.300 libnvrtc.so.10.2.300
libcudart.so.10.2 libcupti.so libmetis_static.a libnppicom_static.a libnppim_static.a libnpps_static.a libnvToolsExt.so
libcudart.so.10.2.300 libcupti.so.10.2 libnppc.so libnppidei.so libnppist.so libnvblas.so libnvToolsExt.so.1
libcudart_static.a libcupti.so.10.2.175 libnppc.so.10 libnppidei.so.10 libnppist.so.10 libnvblas.so.10 libnvToolsExt.so.1.0.0
libcufft.so libcurand.so libnppc.so.10.2.1.300 libnppidei.so.10.2.1.300 libnppist.so.10.2.1.300 libnvblas.so.10.2.3.300 stubs
libcufft.so.10 libcurand.so.10 libnppc_static.a libnppidei_static.a libnppist_static.a libnvgraph.so
libcufft.so.10.1.2.300 libcurand.so.10.1.2.300 libnppial.so libnppif.so libnppisu.so libnvgraph.so.10