Hi @CostGazelle, if you install PyTorch from pip/pip3, it won’t have CUDA enabled. For that, you have to install one of the PyTorch wheels from here: https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048 (those were built with CUDA support)
Related topics
Topic | Replies | Views | Activity | |
---|---|---|---|---|
AssertionError: Torch not compiled with CUDA enabled | 7 | 3993 | January 2, 2024 | |
Failed install pytorch in jetson nano | 9 | 525 | January 16, 2024 | |
Installing Torch and Torchvision with CUDA support on Jetson Nano | 2 | 1413 | March 25, 2024 | |
Running onnx in Jetson nano | 2 | 68 | July 16, 2025 | |
Error with PyTorch 2.0.0+nv23.05 on Jetson Orin Nano | 6 | 50 | May 19, 2025 | |
How to completely execute python code using CUDA | 6 | 1587 | March 7, 2023 | |
Installing pytorch on jetson nano | 7 | 5103 | July 28, 2022 | |
Pytorch to Onnx | 5 | 1038 | October 15, 2021 | |
Jetson Nano Pytorch Wheel files for latest Python 3.8/3.9 versions with CUDA 10.2 support | 5 | 4910 | December 28, 2021 | |
AssertionError: Torch not compiled with CUDA enabled | 8 | 20536 | December 7, 2021 |