How to install torch with cuda

I have jetson orin nano board with jetpack 5.1.2 with ubuntu 20.04. I had installed ultralytics in python version 3.8.10 .I had enable opencv with cuda ,But still when i import torch and cheak cuda available it provides false. Help me with command asap ,I’am not able to get.

Hello,

Your topic will be best served in the Jetson category.

I will move this over for visibility.

Cheers,
Tom

Hi,

Could you check the PyTorch version and share it with us?

>>> import torch
>>> torch.__version__

If it is not our prebuilt (with nv tag), please install it with the following instructions:

Thanks.

I have torch version 2.1.0a0+41361538.nv23.6
when running command
import torch
torch.cuda.is_available()
it shows True, and i install
torchvision 0.16.2
by following this steps PyTorch for Jetson
It works thanks for the support.
But still facing issue how to use deepstream and tensor RT for object detection.

Hi,

Good to know PyTorch and TorchVision work on your environment now.
Please see the following for the examples of Deepstream and TensorRT.

Deepstream: GitHub - NVIDIA-AI-IOT/deepstream_python_apps: DeepStream SDK Python bindings and sample applications
TensorRT: TensorRT/samples/python/efficientdet at release/8.6 · NVIDIA/TensorRT · GitHub

Thanks.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.