hi all im trying to get CUDA on my jetson orin nano when implementing code myself on visual studio code. im trying to download CUDA to run PyTorch and opencv but this website is down. can anyone advice on what I should do, or if there is any other methods that I am able to allow opencv to run with CUDA on jetson orin nano? thanks
Grab from here:
jp6/cu126 index
hello @JimikoSK thank you. i tried downloading but it still says CUDA devices: 0 . i am doing this all in my venv, do i need to do it locally on my device instead? and i am pulling only the opencv-contrib-python wheel, do i need to download any other files or all the files?
@JimikoSK fixed. but when i import torch there seems to be an issue still, i still dont get torch cuda being available.
python3 -c “import torch; print(‘PyTorch:’, torch.version); print(‘CUDA available:’, torch.cuda.is_available())”
PyTorch: 2.9.0+cpu
CUDA available: False
can i get some help for this
It looks like Torch CPU is installed instead of the Jetson Pytorch.
Try:
pip uninstall torch torchvision torchaudio
pip install https://pypi.jetson-ai-lab.io/jp6/cu126/+f/62a/1beee9f2f1470/torch-2.8.0-cp310-cp310-linux_aarch64.whl
pip install https://pypi.jetson-ai-lab.io/jp6/cu126/+f/81a/775c8af36ac85/torchaudio-2.8.0-cp310-cp310-linux_aarch64.whl
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.