Jetson Nano 4gb developer kit Jetpack : 4.5.1 Deepstream : 5.1 Ubuntu : 18.04 Cuda Version : 10.2.89 TensorRt - 7.1.3.1 Python version - 3.10 pyTorch version - 2.0.1 Installed from official website using [image] I am unable to access GPU using pytorch import torch torch.cuda.is_available() …

@Keshav.A uninstall PyTorch first, and you can find more detailed install instructions for PyTorch on Jetson Nano in this topic: [image] PyTorch for Jetson Jetson Nano Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack…

Cuda enable using pytorch

Autonomous Machines Jetson & Embedded Systems Jetson Nano

dusty_nv July 10, 2023, 2:58pm 5

@Keshav.A uninstall PyTorch first, and you can find more detailed install instructions for PyTorch on Jetson Nano in this topic:

If you still are having problems after that, I would recommend trying the l4t-pytorch container which already has PyTorch with CUDA support installed: