AssertionError: Torch not compiled with CUDA enabled

Hi @senasahin, I recommend first uninstalling PyTorch, and then reinstall with one of the wheels that I linked to:

pip3 uninstall torch
sudo pip3 uninstall torch
python3 -c 'import torch'      # this should not be successful

Then install one of the wheels that was built with GPU support from this topic: https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048