Hi @danir1512, you should install one of the PyTorch wheels from here that was built with CUDA enabled (or use the l4t-pytorch container): https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-11-now-available/72048
Regarding the nvcc issue, does it find nvcc if you run make without sudo? I’m curious if the path doesn’t include /usr/local/cuda/bin under the sudo environment.