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 | |
|---|---|---|---|---|
| Torch not compiled with CUDA enabled | 8 | 7208 | April 25, 2023 | |
| Installing pytorch on jetson nano | 7 | 4567 | July 28, 2022 | |
| AssertionError: Torch not compiled with CUDA enabled | 7 | 3636 | January 2, 2024 | |
| Jetson AGX Xavier Jetpack 5 upgraded to CUDA 11.8 can't use torch :( | 5 | 411 | May 13, 2024 | |
| Pytorch with CUDA support Jetson Orin Nano | 4 | 1242 | May 14, 2024 | |
| AssertionError:Torch not compiled with CUDA enabled :trt_pose | 7 | 696 | December 26, 2023 | |
| Torch not compiled with CUDA enabled error | 4 | 890 | November 15, 2023 | |
| I want to change yolov8 onnx to engine(tensorrt) for using DLA ,but faild Torch not compiled with CUDA enabled | 2 | 128 | July 4, 2024 | |
| Pytorch with CUDA for Jetpack 5.1.4 | 4 | 265 | January 13, 2025 | |
| Torch not compiled with cuda enabled over Jetson Xavier Nx | 2 | 3619 | October 18, 2021 |